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

假设我们的矩阵是：

#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列的最大值的行号。谢谢可以通过以下方式检索行号：

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

结果如下：

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

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