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

12 浏览
0 Comments

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

我可以找到很多排列组合的问题,但找不到这个(相当简单的)问题的答案:如何以最pythonic的方式找到numpy数组中特定列的最大值?

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

我想要的是第一列和第二列的最大值(这些是x,y坐标,我最终需要每个形状的高度和宽度),所以最大的x坐标是10,最大的y坐标是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])

这个方法可行,但似乎不是最好的方法。

有什么建议吗?

0