查找条件为真的第n个坐标

Question

我有一个多维的numpy数组，我想找到条件为真的第n个位置。

如果我想找到A为零的第n个位置，理论上我可以做到

coords = np.array(np.where(A==0))
coord = coords[:, n]

然而，这非常慢，因为首先检查A中所有条目的条件（A == 0），然后为这些条目生成所有坐标，然后丢弃除第n个坐标之外的所有坐标。

有更快的方法吗？