Finding the element in one array corresponding to the maximum value in another

问题

How can I align the 4d z array and the 4d QCLOUD array and to then find out the z value of when QCLOUD max occurs?

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

回答1:

Ignoring the fact that (np.array(z.shape) > np.array(qcloud.shape)).any(), you want argmax:

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

来源：https://stackoverflow.com/questions/37446740/finding-the-element-in-one-array-corresponding-to-the-maximum-value-in-another