如何在numpy数组列中找到最大值?
- 论坛
- 如何在numpy数组列中找到最大值?
12 浏览
如何在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])
这个方法可行,但似乎不是最好的方法。
有什么建议吗?