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

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']]

帮忙找错误。 谢谢。

Answer 1

一种方法是：

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

或者，如果您希望项目为int ：

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

Answer 2

代码：

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

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