numpy在數組中找到最大數字

我有一個形狀不規則的數組(1, 7, 3)我想查找3列中具有最高元素的行。例如：232是第三列中最大的，所以它應該輸出[196 228 232]我該怎麼做？我試圖np.argmax但未能numpy在數組中找到最大數字

下面是一個例子數組：

[[[218 204 204] 
[344 194 31] 
[284 140 108] 
[196 228 232] 
[324 196 28] 
[224 228 57] 
[174 250 144]]]

2017-02-20 Wideem

也許這個鏈接可以幫助http://stackoverflow.com/questions/12403238/maximum-values-along-axis-of-numpy-ndarray的 – phunsukwangdu

可能的複製[最大Numpy ndarray軸的值？]（http://stackoverflow.com/questions/12403238/maximum-values-along-axis-of-numpy-ndarray） – Evert

a.max（axis = 2）返回'[218 344 284 232 324 228 250]'這是每行的最大值，並且我需要具有最大行[2]元素的元素，如問題 – Wideem

argmax這裏是正確的想法。讓我們一步一步來做。

row_nr = np.argmax(data[0, :, 2])

這選擇第三列並找到最大值的索引。它仍然選擇該行：

data[0, row_nr, :]

2017-02-20 23:46:52

感謝您花時間閱讀和回答我的問題。它現在非常有意義 – Wideem

回答