从CSV获取列和关联行的最大值

时间:2016-12-20 19:34:15

标签: python max

我每年都有犯罪统计信息的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:]

由于

1 个答案:

答案 0 :(得分:2)

您需要使用np.argmax()功能。这就是你如何做到的,

num_crimes = crime_code[:,1].astype(int)
codes = crime_code[:,0].astype(int)
code = codes[np.argmax(num_crimes)]