我有一个形状为(1, 7, 3)的numpy数组我希望在3列中找到具有最高元素的行。例如：232是第3列中最大的，所以它应该输出[196 228 232]我该怎么做？我试过np.argmax但是失败了
这是一个示例数组：
[[[218 204 204]
[344 194 31]
[284 140 108]
[196 228 232]
[324 196 28]
[224 228 57]
[174 250 144]]]
答案 0 :(得分：1)
argmax这是正确的想法。让我们一步一步来做。
row_nr = np.argmax(data[0, :, 2])
这将选择第三列并查找最大值的索引。 它仍然是选择这一行：
data[0, row_nr, :]