King Andy King Andy - 6 months ago 10
Python Question

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)

Answer

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

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