在numpy数组中查找超出以前值的值的索引

Question

我有一个矮小的数组：

>>> n1 = np.array([[1, 4], [1, 5], [2, 4], [7, 2], [1, 3], [4, 7], [2, 9]])
>>> n1
array([[1, 4],
       [1, 5],
       [2, 4],
       [7, 2],
       [1, 3],
       [4, 7],
       [2, 9]])

我正在寻找一种方法，在第一列中查找值出现的索引，该索引的值大于值的出现。在这种情况下，我想：

array([4, 6])

因为值1小于7，值2小于4(全部在第0列中)

有一种很好的毕达通的方法吗？

Answer 1

您可以在第一列上使用numpy.diff并查找值为负值的位置。

np.where(np.diff(n1[:, 0]) < 0)

如果需要，添加1以调整索引。