如果有一个形状为(9,1,3)的数组。
array([[[ 6, 12, 108]],
[[122, 112, 38]],
[[ 57, 101, 62]],
[[119, 76, 177]],
[[ 46, 62, 2]],
[[127, 61, 155]],
[[ 5, 6, 151]],
[[ 5, 8, 185]],
[[109, 167, 33]]])
我想找出第三个维度的 argmax 索引, 在这种情况下它应该是185,因此索引为7。
我猜解决方案与重塑(reshaping)有关,但我理解不了。感谢任何帮助!
np.argmax(yourarray[:, 0, 2])
。 - Paul Panzernp.argmax(np.max(arr, 2))
怎么样? - kmario23np.unravel_index(np.argmax(arr), arr.shape)
?只要能得到一个包含数字七的结果。 - Paul Panzer