我可以找到这个问题的很多排列,但不是这个(相当简单)的排列:如何找到numpy数组的特定列的最大值(以最pythonic的方式)?
a = array([[10, 2], [3, 4], [5, 6]])
我想要的是第一列和第二列中的最大值(这些是x,y坐标,最终我需要每种形状的高度和宽度),因此max 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])
哪个可行,但似乎不是最好的方法。
有什么建议吗?
只需打开列表包装即可:
In [273]: xmax, ymax = a.max(axis=0)
In [274]: print xmax, ymax
#10 6
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句