Numpy: Получение строки номер элемента, который имеет максимальное значение при n'th столбца в матрице

Question

Предположим, что наша матрица будет:

#alternatives 
a1="4 1 0 1 0" 
a2="5 1 0 1 0" 
a3="6 0 0 1 0" 
a4="7 1 1 0 0" 
a5="6 0 0 1 0" 
a6="3 0 1 1 0" 
a7="7 1 1 0 1" 
import numpy as np 
dm=np.matrix(a1+";"+a2+";"+a3+";"+a4+";"+a5+";"+a6+";"+a7) 

Я хочу получить строки номер из максимального значения из 5 -м column.Thanks

Answer 1

номер строки можно получить таким путем:

sq=dm[:,4] 
print(np.where(sq==np.max(sq))) 

Результат выглядит так:

(matrix([[6]]), matrix([[0]])) 

6 - номер строки элемента, который имеет максимальное значение в 5-м столбце.