從CSV獲取列和關聯行的最大值

Question

我有一個每年犯罪統計的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:] 

感謝

Answer 1

您將需要使用np.argmax()功能。這就是你如何做到這一點，

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