问题描述
我正在处理的项目在其数据目录中有 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 在同一行上处理多个运算符的确切顺序,所以对此的简短解释也很好。
解决方法
列表组合的等效代码:
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
如果您有任何疑问或对答案不满意,请告诉我