嵌套Python列表推导式来构造列表列表

Question

我是一个 python 新手，在摸索嵌套列表理解时遇到了麻烦。我正在尝试编写一些代码来读取文件并为每行的每个字符构建一个列表。

因此，如果文件包含

xxxcd
cdcdjkhjasld
asdasdxasda

结果列表将为：

[
['x','x','x','c','d']
['c','d','c','d','j','k','h','j','a','s','l','d']< br> ['a','s','d','a','s','d','x','a','s','d','a']
]

我已经编写了以下代码，它可以工作，但我有一种挥之不去的感觉，我应该能够编写一个嵌套列表理解来用更少的代码行来完成此操作。任何建议将不胜感激。

data = []
f = open(file,'r')
for line in f:
    line = line.strip().upper()
    list = []
    for c in line:
        list.append(c)
    data.append(list)

Answer 1

这应该会有所帮助（您可能需要使用它来删除换行符或按照您想要的方式格式化它，但基本思想应该可行）：

f = open(r"temp.txt")
[[c for c in line] for line in f]

Answer 2

在您的情况下，您可以使用 list 构造函数来处理内部循环，并使用列表理解来处理外部循环。类似于：

f = open(file)
data = [list(line.strip().upper()) for line in f]

给定一个字符串作为输入，列表构造函数将创建一个列表，其中字符串的每个字符都是列表中的单个元素。

列表理解在功能上等同于：

data = []
for line in f:
    data.append(list(line.strip().upper()))

Answer 3

这是列表理解的一个级别。

data = []
f = open(file,'r')

for line in f:
    data.append([ch for ch in line.strip().upper()])

但我们可以一次性完成整个事情：

f = open(file, 'rt')
data = [list(line.strip().upper()) for line in f]

这是使用 list() 将字符串转换为单字符字符串列表。我们还可以使用嵌套列表推导式，并将 open() 内联：

data = [[ch for ch in line.strip().upper()] for line in open(file, 'rt')]

不过，在这一点上，我认为列表推导式降低了正在发生的事情的易读性在。

对于复杂的处理，例如列表中的列表，您可能需要对外层使用 for 循环，对内层循环使用列表理解。

此外，正如 Chris Lutz 在评论中所说，在这种情况下，确实没有理由将每一行显式分割为字符列表；您始终可以将字符串视为列表，并且可以对字符串使用字符串方法，但不能对列表使用字符串方法。 （好吧，您可以使用 ''.join() 将列表重新连接回字符串，但为什么不将其保留为字符串呢？）

Answer 4

data = [list(line.strip().upper()) for line in open(file,'r')]

Answer 5

字符串和字符列表之间唯一真正显着的区别是字符串是不可变的。您可以像列表一样迭代和切片字符串。将字符串作为字符串处理要方便得多，因为它们支持字符串方法，而列表则不支持。

因此，对于大多数应用程序，我不会费心将 data 中的项目转换为列表；我只是这样做：

data = [line.strip() for line in open(filename, 'r')]

当我需要将 data 中的字符串作为可变列表进行操作时，我会使用 list 来转换它们，然后 join将它们放回去，例如：

data[2] = ''.join(sorted(list(data[2])))

当然，如果您要对这些字符串做的只是修改它们，那么继续，将它们存储为列表。

Answer 6

>>> f = file('teste.txt')
>>> print map(lambda x: [c for c in x][:-1], f)
[['x', 'x', 'x', 'c', 'd'], ['c', 'd', 'c', 'd', 'j', 'k', 'h', 'j', 'a', 's', 'l', 'd'], ['a', 's', 'd', 'a', 's', 'd', 'x', 'a', 's', 'd']]

Answer 7

首先，您可以将 line.strip().upper() 部分与外部 for 循环结合起来，如下所示：

for line in [l.strip().upper() for l in f]:
    # do stuff

然后您可以将字符迭代转换为列表理解，但它不会更短或更清晰。在那里做你所做的事情的最巧妙的方法是这样的：

list(someString)

因此你可以这样做：

data = [list(l.strip().upper()) for l in f]

但我不知道它是否很好地表达了你的意图。错误处理也是一个问题，如果途中出现问题，整个表达式就会死掉。

---

如果您不需要将整个文件和所有行存储在内存中，您可以将其放入生成器表达式中。当处理大文件并且您一次只需要处理一个块时，这非常有用。生成器表达式使用括​​号，如下所示：

data = (list(l.strip().upper()) for l in f)

data 将成为一个生成器，为文件中的每一行运行表达式，但仅当您对其进行迭代时；将其与列表理解进行比较，列表理解将在内存中创建一个巨大的列表。请注意，data 不是列表，而是生成器，更类似于 C++ 中的迭代器或 C# 中的 IEnumerator。

生成器可以很容易地输入到列表中： list(someGenerator) 这会在某种程度上违背目的，但有时是必要的。