Python：具有列表列表的列表理解

Question

我将如何在列表理解中编写以下代码？grid = open('some_file.txt', 'r')lines = [line.strip('\n') for line in grid]list_of_lists = []for line in lines:    elms = [int(elm) for elm in line.split(' ')]    list_of_lists.append(elms)我的文件如下所示：3 8 6 9 44 3 0 8 62 8 3 6 93 7 9 0 3意思就是：grid = '3 8 6 9 4\n4 3 0 8 6\n2 8 3 6 9\n3 7 9 0 3'

Answer 1

在这里尝试一下，首先拆分每一行，您将获得一个数字列表作为字符串，因此map可以使用函数将其更改为int：

with open('file.txt', 'r') as f:

    k = [list(map(int,i.split())) for i in f.readlines()]

    print(k)

Answer 2

你并不需要应用str.strip和str.split独立。相反，将它们组合在一个操作中。列表推导式是通过定义一个列表元素，然后在循环上进行迭代来构建的for。

另请注意，str.strip不带参数将与\n空格一样处理。同样，str.split没有参数的情况下也会被空格分隔。

from io import StringIO

x = StringIO("""3 8 6 9 4

4 3 0 8 6

2 8 3 6 9

3 7 9 0 3""")

# replace x with open('some_file.txt', 'r')

with x as grid:

    list_of_lists = [[int(elm) for elm in line.strip().split()] for line in grid]

结果：

print(list_of_lists)

[[3, 8, 6, 9, 4],

 [4, 3, 0, 8, 6],

 [2, 8, 3, 6, 9],

 [3, 7, 9, 0, 3]]

使用内置功能，使用起来效率更高map：

list_of_lists = [list(map(int, line.strip().split())) for line in grid]