查找给定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]