通过numpy中行的最大值查找列索引

Question

给出了以下numpy：

myArray = np.array([[ 3, 8, 0, 0],
                    [ 9, 7, 8, 1],
                    [ 3, 8, 9, 7]])

需要找到一行中每个最大值的列索引。

即根据上面的数据应该返回如下列表：

[1,0,2]

Answer 1

您可以使用带有axis = 1属性的np.argmax函数，这意味着查找行索引。

max_index_col = np.argmax(myArray, axis=1).tolist()

输出

>> max_index_col 
[1, 0, 2]