在同一行上生成带有for和if的筛选列表

Question

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

定义目标目录：

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

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