使用 for 和 if 在同一行生成过滤列表

Question

我正在处理的项目在其数据目录中有 3 种不同类型的文件夹：绿色、黄色和红色。每种数据类型都有许多文件夹，它们的名称以各自的颜色开头，如：“RED_folder1”。我发现了以下代码，它具有生成包含特定类型的所有文件夹的列表的 objective。

定义目标目录：

directory = '/project/data'

生成红色类型的文件夹列表：

red_folders = [k for k in os.listdir(directory) if os.path.isdir(os.path.join(directory,k)) and 'RED_' in k[:len('RED_')]]

我觉得这段代码很奇怪，因为它在 os.listdir(directory) 中创建了符合给定标准的 k 个条目的列表，但标准本身也取决于 k。这行代码是按什么顺序处理的？假设创建整个 k in os.listdir(directory) 列表然后考虑 if 语句并丢弃不适当的条目是否正确？

我的困惑可能来自于不知道 python 在同一行中处理多个运算符的确切顺序，所以对此进行简短解释也很好。

Answer 1

列表组合的等效代码：

for k in os.listdir(directory):
     if os.path.isdir(os.path.join(directory,k)) and 'RED_' in k[:len('RED_')]:
           red_folders.append(k)

首先生成一个目录列表。然后 var k 遍历序列的每个元素。执行 If 语句。如果发现条件为真，则附加 k。现在，在您的列表组合中，相同的机制正在发挥作用。唯一的区别是生成一个全新的列表而不是 red_folders.append(k)，然后将其命名为 red_folders.

red_folders=new_list_created_by_list_composition

如果您有任何疑问或对答案不满意，请告诉我

Generating a filtered list with for and if on the same line