Поиск элемента в одном массиве, соответствующего максимальному значению в другом

Как я могу выровнять массив 4d z и массив 4d QCLOUD и затем узнать значение z, когда происходит QCLOUD max?

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

One Solution collect form web for “Поиск элемента в одном массиве, соответствующего максимальному значению в другом”

Игнорируя тот факт, что (np.array(z.shape) > np.array(qcloud.shape)).any() , вы хотите argmax :

 idx = np.argmax(qcloud) result = z[tuple(idx)] 
Python - лучший язык программирования в мире.