查找嵌套列表中不同子列表的相同索引中的元素之间的运行差异

Question

我有一个嵌套列表，我必须通过它迭代和比较（并随后操作）每个子列表的相同位置/索引中的元素。 例如，这是一个最小可重现程序：

a = [[1,2,3], [1,4,7], [2,0,8], [5,1,1]] #nested list

b = [[1,2,3], [0,2,4], [1,-4,1], [3,1,-7]] #desired output

我基本上需要创建一个新的嵌套列表，其中解决方案中的每个后续子列表都是原始嵌套列表中相同位置/索引中的子列表与原始嵌套列表中之前位置/索引中的子列表之间的差异。 即，b中的第一个子列表只是a中的第一个子列表，b中的第二个子列表是a中的第二个子列表减去a中的第一个子列表，b中的第三个子列表是a中的第三个子列表减去a中的第二个子列表, ...., b中的第n个子列表是a中的第n个子列表减去a中的第n-1个子列表。

我需要以自动和迭代的方式执行此操作，因为我在上面提供的这些列表不是真正的列表——真正的嵌套列表包含数千个子列表。 我是 python 的新手，所以如果这很简单，我深表歉意 - 如果是这种情况，如果您能指导我找到解决方案，我将非常感激。

Answer 1

np.diff是为它设计的：

import numpy as np
b = np.r_[[a[0]], np.diff(a, axis=0)]

请注意，output 是numpy数组。 它比标准的 Python 迭代快得多。

Answer 2

import numpy as np

a = [[1,2,3], [1,4,7], [2,0,8], [5,1,1]] 
b = [a[0], *[list(i) for i in np.diff(a, axis=0)]]

或者，如果您想使用纯 Numpy 方法，则

b = np.ndarray.tolist(np.r_[[a[0]], np.diff(a, axis=0)])