如何在numpy数组列中找到最大值？

Question

我可以找到这个问题的很多排列，但不是这个（相当简单）的排列：如何找到numpy数组的特定列的最大值（以最pythonic的方式）？

a = array([[10, 2], [3, 4], [5, 6]])

我想要的是第一列和第二列中的最大值（这些是x，y坐标，我最终需要每个形状的高度和宽度），所以max x coordinate是10，max y coordinate是6。

我试过：

xmax = numpy.amax(a,axis=0)
ymax = numpy.amax(a,axis=1)

但是这些让步

array([10, 6])
array([10, 4, 6])

…不是我想的那样。

我的解决方案是使用切片：

xmax = numpy.max(a[:,0])
ymax = numpy.max(a[:,1])

但似乎不是最好的方法。

建议？

Answer 1

打开清单：

In [273]: xmax, ymax = a.max(axis=0)

In [274]: print xmax, ymax
#10 6