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

回答