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

问题描述

我正在处理的项目在其数据目录中有 3 种不同类型的文件夹:绿色、黄色和红色。每种数据类型都有许多文件夹,它们的名称以各自的颜色开头,如:“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 在同一行上处理多个运算符的确切顺序,所以对此的简短解释也很好。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)