python - 如何使用Python中的列表理解从一个元素中获取多个元素？

Question

我的标题可能不太清楚，抱歉，不容易解释。但举个例子会更容易理解。

a = [1, 4, 7, 10]

b = [x-1, x, x+1 for x in a]

>> expected result: [0,1,2,3,4,5,6,7,8,9,10,11]

当然，它会引发错误。

是否可以使用列表理解来获得此结果？

我知道我可以使用 b = sum([[x-1, x, x+1] for x in a], []) 否则，但它的可读性较差。

Answer 1

您可以使用两个 for 循环来实现这一点:

b = [x+i for x in a for i in (-1,0,1)]

现在b包含:

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]