python - Numpy 三列之间的最大参数项数，输出为数组

Question

我想制定一个数组，它是 3 列之间的最大值(项目编号不是值)。

例如

In:  arr=([(1,2,3,4), (4,5,16,0), (7,8,9,2)]) # maximum of columns 0, 1, 2, 3
Out: array([2,2,1,0])  # As: 7 > 4 > 1, 8 > 5 > 2, 16 > 9 > 3, and 4 > 2 > 0

当前(非工作解决方案):

np.argmax([arr['f0'], arr['f1'], arr['f2']])

Answer 1

您可以在numpy.argmax中指定axis键，它独立地对numpy数组的指定轴进行操作。在您的情况下，您希望通过查找每列最大值的索引来单独操作每列，因此请指定 axis=0。以下是在 IPython 中给定数据的示例运行:

In [10]: import numpy as np

In [11]: arr=np.array([(1,2,3), (4,5,16), (7,8,9)])

In [12]: np.argmax(arr, axis=0)
Out[12]: array([2, 2, 1])

上面的示例是您编辑帖子之前的示例。使用编辑中的新数据，以下是运行示例:

In [13]: arr=np.array([(1,2,3,4), (4,5,16,0), (7,8,9,2)])

In [14]: np.argmax(arr, axis=0)
Out[14]: array([2, 2, 1, 0])

有关numpy.argmax的更多信息可以在这里找到:http://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html