Python:os.listdir替代/某些扩展

Question

是否可以使用os.listdir命令查看具有特定扩展名的文件？我希望它能够工作,因此最终可能只显示带有.f的文件或文件夹.我检查了文档,发现什么都没有,所以不要问.

Answer 1

glob 擅长这个:

import glob
for f in glob.glob("*.f"):
    print f

Answer 2

不要问什么？

[s for s in os.listdir() if s.endswith('.f')]

如果要查看扩展列表,可以进行明显的概括,

[s for s in os.listdir() if s.endswith('.f') or s.endswith('.c') or s.endswith('.z')]

或者另一种方式是写一点:

[s for s in os.listdir() if s.rpartition('.')[2] in ('f','c','z')]