我每年都有犯罪统计信息的CSV文件,我需要使用Python来显示每个。在我的第一栏中,我有犯罪代码。右边的列是每年犯罪案件的数量。我试图找出每年的最大值,但也返回犯罪代码。
首先我定义了:
Year_2003 = crime_code[:, 1].astype(int)
max_crime_2003 = np.max(Year_2003)
print max_crime_2003
我知道这确实给了我这个专栏的最大价值。任何人都可以帮我告诉我如何返回第一列中的犯罪代码? 我搜索了论坛并在嵌套列表中找到了一些东西,所以我还在我的代码中包含了以下内容:
nested_list = [row for row in readCSV]
np_list = np.array(nested_list, dtype = str)
crime_code = np_list[1:]
由于
答案 0 :(得分:2)
您需要使用np.argmax()
功能。这就是你如何做到的,
num_crimes = crime_code[:,1].astype(int)
codes = crime_code[:,0].astype(int)
code = codes[np.argmax(num_crimes)]