在一个数组中查找与另一个数组中的最大值对应的元素

Question

如何对齐4d z数组和4d QCLOUD数组，然后找出QCLOUD max出现时的z值？

print(z.shape)
print(qcloud.shape)
out: (6, 100, 128, 128)
(6, 99, 128, 128)

Answer 1

忽略(np.array(z.shape) > np.array(qcloud.shape)).any()，您想要argmax：

这一事实

idx = np.argmax(qcloud)
result = z[tuple(idx)]