Найти соответствующие [x, y] координаты для заданного значения координаты z

Question

У меня есть 3D-массив, и я должен был найти наибольшую Z-координату в этом массиве.После этого мне нужно найти соответствующие значения координат X и Y на основе Z-координаты.Как я могу быстро добиться этого с помощью numpy?

Что я сделал: я использовал argsort, чтобы сначала отсортировать заданный 3D-массив, затем использовал np.max (массив), чтобы найти наибольшую Z-координату.Я не знаю, как еще продолжить.Может ли numpy.where быть полезным здесь?

Спасибо!

Answer 1

То, что вы ищете, это numpy argmax

быстрый пример:

import numpy as np

data = np.random.rand(5,3)
print data
ind = np.argmax(data[:,2])
print data[ind, :]

выходы

[[0.92037795 0.59469121 0.02956843]
 [0.82881039 0.23272832 0.97275488]
 [0.98418468 0.45699429 0.44662552]
 [0.62519115 0.16637013 0.40433299]
 [0.98272718 0.01467489 0.57442259]]

[0.82881039 0.23272832 0.97275488]