2012-09-13 33 views

回答

2

如果选中的np.max的文件,你会看到它需要一个axis参数:

a.max(axis=0) 

但是,这不会帮助你的呢。然而,有,让你的最大的指数沿给定轴的功能argmax

a.argmax(axis=...) 

那么,让我们看看你的第一个(5,5)数组:它的a[...,0]。您可以使用a[...,0].max(axis=1)(或0)查找每行(或多列)的最大值位置,并使用它来查找另一侧的值。

+0

+1如何获得最大值的“层数”? – oops

+0

谢谢!实际上,argmax(axis = 0)返回我需要的数组 – oops