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

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个匹配的内容，而不会创建多个匹配文件名的列表。