如何使用argmax沿轴获取numpy 2D数组每一列的最大值？

Question

例如，我有以下2D数组：

array([[ 5.59635947,  1.42474555,  1.56519762],
       [ 6.16476541,  6.12324772,  5.32735447],
       [ 8.60258444,  7.16592582,  1.49555662],
       [ 0.63983973,  5.50249666,  3.52246942],
       [ 5.34002941,  4.87065573,  9.80725886]])

我使用argmax沿axis = 0获取最大元素在每一列中的位置：

i = np.argmax(c,axis=0)
array([2, 2, 4])

好吧，我已经计算出了指数。如何获得每列的最大元素？

当我编写以下代码时：

dd_m= np.zeros(i.shape)
for k in np.arange(i.size):
    dd_m[k] = c[i[k],k]

我得到正确的答案，即：

 array([ 8.60258444,  7.16592582,  9.80725886])

是否有更有效的方法或pythonic方法？

Answer 1

尝试：

c.max(axis=0)

或者：

np.max(c, axis=0)

或者如果您已经计算出指数：

c[i, np.arange(c.shape[1])]

其中c是矩阵，i是找到的最大值索引。