Python:列表理解列表列表

Question

我有一个列表列表,并希望使用列表推导将函数应用于列表列表中的每个元素,但是当我这样做时,我最终得到一个长列表而不是列表列表.

所以我有

x = [[1,2,3],[4,5,6],[7,8,9]]
[number+1 for group in x for number in group]
[2, 3, 4, 5, 6, 7, 8, 9, 10]

但我想得到

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

我该怎么做呢？

Answer 1

用这个:

[[number+1 for number in group] for group in x]

如果您知道地图,请使用此选项:

[map(lambda x:x+1 ,group) for group in x]