Python模块之glob

介绍

glob模块可以按指定路径得到所有符合条件的文件的完整路径。

使用

import glob

pattern = 'data/*/*'
file_path = glob.glob(pattern)
# 得到的是一个list

for path in file_path:
        print(path)

# data/train/001
# data/train/002
# ...
# data/dev/001
# data/dev/002
# ...
# data/test/001
# ...

如果想得到一个generator，以减少内存，可以使用 iglob（）函数

files = glob.iglob(pattern)
type(files)