Numpy：在矩阵的第n列获取具有最大值的元素的行号

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列最大值的行号。谢谢

Answer 1

可以通过这种方式检索行号：

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

结果如下：

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

6是在第5列具有最大值的元素的行号。