循环遍历目录中的固定数量的文件

Question

如何在目录中循环固定数量的文件glob.glob？如果该目录中有超过x个文件,我只想循环遍历x然后退出循环.我该怎么做呢？

Answer 1

使用itertools.islice()和glob.iglob()有效限制可以循环的结果数量:

from itertools import islice
import glob

x = 1000
limited_files = islice(glob.iglob('pattern.*'), x)
for filename in limited_files:
    #  handle filename

即使在目录中有太多匹配太多的匹配,你也永远不会处理超过前1000个匹配,而不创建具有匹配文件名的多个列表.