如何互相减去numpy数组中的相邻向量

时间:2019-03-01 19:55:37

标签: python numpy

我的预期结果如下:

array = [[2,3,4], [1,2,4]]

输出:

[1, 1, 0]  # [2-1, 3-2, 4-4]

我尝试通过枚举并让索引减去的方式来做到这一点,但运气不好:

for i, k in enumerate(array):
    for j in k:
        return(j[i+1] - j[i])

哪个给我:

  

IndexError:标量变量的索引无效。

1 个答案:

答案 0 :(得分:0)

这有效:

result = [(i-j) for (i,j) in zip(*array)]

输出:

print (result)
[1, 1, 0]

说明:

zip(*array)等同于元组列表[(2,1), (3,2), (4,4)]