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

Question

我想从输入数字生成嵌套的 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']]

帮助找出错误。谢谢。

标签： pythonlistlist-comprehension

Answer 1

一种方法是：

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

或者，如果您希望这些项目是int：

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