Получите максимальное значение столбца и соответствующей строки из 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)]