python - 查找 numpy 数组中超过先前值的值的索引

Question

我有一个 numpy 数组:

>>> 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 列)

有没有一种很好的Pythonic方式来做到这一点？

Answer 1

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

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

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