如何在不制作产品的情况下进行嵌套列表理解？

Question

在Python中，我想列出一组目录中的所有文件。我最想要的是一份清单。但最多我设法制作一个嵌套列表：

pics = os.path.expanduser('~/Pictures')
all_pics = [(d, os.listdir(d)) for d in os.listdir(pics)]

结果：

[('folder1', ['file1', 'file2', ...]), ('folder2', ['file1', ...]), ...]

我想要什么：

[('folder1' 'file1'), ('folder1', 'file2'), ..., ('folder2', 'file1'), ...]

我想要得到的是一个简单的普通列表，无论什么（可以是元组），只是为了它没有嵌套的东西，并且我不需要在解析它的代码中嵌套循环。

我怎样才能用列表理解来做到这一点？这样做会得到 2 组（目录名和文件名）的乘积，这是错误的：

[(d, f) for f in os.listdir(os.path.join(pics, d)) for d in os.listdir(pics)]

Answer 1

您的 for 循环顺序错误。它应该是

all_pics = [(d, f)
            for d in os.listdir(pics)
            for f in os.listdir(os.path.join(pics, d))]

最外层循环在前，最内层循环在最后。

我想知道为什么你没有得到 d 的 NameError 。

Answer 2

使用 os.walk 是一个更好的主意：

all_pics = [(dir,file)
            for dir,sub,files in os.walk(pics) 
            for file in files]

那么为什么它是一个更好的主意呢？

• 它也适用于更深的树（subdir/subdir/pic.jpg），
• 如果您将文件放入~/Pictures（如果您调用>os.listdir 在文件上，你会得到OSError：[Errno 20]不是目录）
• 它更简单：os.walk为你做遍历，所有剩下的就是格式化输出

Answer 3

列表推导式作为从一个列表到另一个列表的映射。正如另一张海报指出的那样，您可以通过嵌套它们来做到这一点，但您确定您真的想要这样做吗？在大多数情况下，您应该优先考虑可读性 - 而且我不认为嵌套理解很容易理解。

What is wrong with? It might be marginally slower - but will you notice?

files = []
for d in os.listdir(pics):
    for f in os.listdir(os.path.join(pics, d)):
        files.append((d, f))

如果您不想生成列表，除非绝对必要，您可以使用生成器：

def get_file_mapping():
    for d in os.listdir(pics):
        for f in os.listdir(os.path.join(pics, d)):
            yield (d, f)
files = list(get_file_mapping())

Answer 4

您只需要展平您的输出列表：

pics = os.path.expanduser('~/Pictures')
all_pics = reduce(lambda xs,ys: xs+ys, [(d, os.listdir(d)) for d in os.listdir(pics)])

Answer 5

l = [ [(d, f) for f in os.listdir(os.path.join(pics, d))] for d in os.listdir(pics) ]
l = sum(l, []) # flatten list of lists => list