沿Numpy ndarray軸的最大值？

Question

numpy_array http://www.freeimagehosting.net/newuploads/klipa.jpg

恐怕所以我畫it.Anyway的素描時，我能不能說明問題，我需要的是找到一個numpy的ndarray沿零軸的最高值，iearray。形狀（5,5,3）及其對應的「圖層編號」，並使用「圖層編號」創建一個形狀爲（1,5,3）的新2D數組。希望我在此給出一個明確的描述..非常感謝。

Answer 1

如果選中的np.max的文件，你會看到它需要一個axis參數：

a.max(axis=0) 

但是，這不會幫助你的呢。然而，有，讓你的最大的指數沿給定軸的功能argmax：

a.argmax(axis=...) 

那麼，讓我們看看你的第一個（5,5）數組：它的a[...,0]。您可以使用a[...,0].max(axis=1)（或0）查找每行（或多列）的最大值位置，並使用它來查找另一側的值。