查找給定z坐標值的相應[x，y]坐標

Question

我有一個3D陣列，我將在該陣列中找到最大的Z坐標。 之后，我需要根據Z坐標找到相應的X和Y坐標值。 如何通過numpy快速實現？

我所做的：我使用argsort首先對給定的3D數組進行排序，然后使用np。 max（array）查找最大的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]