从下一行的第一个元素中减去当前行的第二个元素

Question

我有6x2的numpy数组。 我想从下一行第一个元素中减去当前行第二个元素，并获取新列。 目前，我正在使用for循环来做到这一点。 例如：

from numpy import array
a = array([[1,3],
           [2,5],
           [6,7],
           [8,9],
           [5,6],
           [6,9]])
k = []
for i in range(a.shape[0]-1):
    k.append(a[i][1]-a[i+1][0])
array(k)

Output : [1, -1, -1, 4, 0]

如何使用numpy获得相同的输出？

Answer 1

切片和减去-

a[1:,0] - a[:-1,1]

样品运行-

In [303]: a
Out[303]: 
array([[1, 3],
       [2, 5],
       [6, 7],
       [8, 9],
       [5, 6],
       [6, 9]])

In [304]: a[1:,0] - a[:-1,1]
Out[304]: array([-1,  1,  1, -4,  0])

因为我们只有两列，所以另一种方法/技巧是在扁平化版本上使用differention，然后从第二个元素开始进入其他每个元素-

In [308]: np.diff(a.ravel())[1::2]
Out[308]: array([-1,  1,  1, -4,  0])