Question

我有一个numpy数组，我想在每行中取3个数字，从下一行中减去它们，然后将这些值存储在另一个数组中。

类似

for i in array:
    a = i - i+1

我知道这是非常错误的，但是至少这给出了我想要的想法。

很明显，i + 1只会得到值+1，然后我所拥有的就是a = 1,1,1

当我说i + 1时，我的意思是下一行。

例如：

input = np.array([[4,4,5], [2,3,1],[1,2,0]])
output =  np.array([2,1,4],[1,1,1]) etc....

在数千行上重复执行此操作的最佳方法是什么？

Answer 1

IIUC，而不是循环，您可以使用np.roll将数组向上移动1，从原始输入中减去该数组，然后取除最后一个数组之外的所有结果数组（因为从数组中不减去任何内容）最后一个数组）：

>>> inp = np.array([[4,4,5], [2,3,1],[1,2,0]])
>>> inp
array([[4, 4, 5],
       [2, 3, 1],
       [1, 2, 0]])
>>> (inp - np.roll(inp,-1,axis=0))[:-1]
array([[2, 1, 4],
       [1, 1, 1]])

或者，更直接的方法就是使用numpy索引：

>>> inp[:-1] - inp[1:]
array([[2, 1, 4],
       [1, 1, 1]])

循环以在numpy数组中的i + 1上执行操作

1 个答案: