在同一行的“for”中编译“if”

不能合并 if 陈述成 for 陈述。但是,可以使用生成器 表达 在一个 对于 陈述。

#            V --- generator expression ---------------------------------------------- V
for files in (files for files in os.listdir(path) if files.endswith("geneclusters.txt")):
    ...

请注意,没有性能优势,事实上这通常是较慢的。最重要的是,它通常不太可读。

做 不 使用列表理解,以 [] 而不是 () ,对于此类用例。它在内存中创建整个中间列表。

你可以使用列表理解:

for files in [f for f in os.listdir(path) if f.endswith("geneclusters.txt")]:

当然(使用列表理解),这取决于您接下来想做什么:

for file in [f for f in os.listdir(path) if f.endswith("geneclusters.txt")]: