如何使用列表推导创建嵌套列表

时间:2019-12-28 15:37:51

标签: python list list-comprehension

我想根据输入数字生成一个嵌套的2级列表。该行的结尾是“输入”。

a = [[i for i in input().split()] for i in input().split (sep = '\ n')]

在这种情况下,这仅占用第二行。 例如:

1 2 3
4 5 6
7 8 9

它将输出如下:

[['4', '5', '6']]

我想要这样的最终结果:

[['1', '2', '3'], ['4', '5', '6'], ['7', '8', '9']]

帮助查找错误。谢谢。

2 个答案:

答案 0 :(得分:1)

一种方法是:

[x.split() for x in data.splitlines()]

或者如果您希望这些项目成为int

[[int(x) for x in x.split()] for x in data.splitlines()]

答案 1 :(得分:0)

代码:

a = [[j for j in i.split()] for i in input().split(sep = '\n')]

您希望内部列表枚举外部列表的元素。 此外,删除多余的空间。