有没有更好的方法来使用glob。Glob在python中获取多个文件类型的列表，如.txt， .mdown和.markdown?现在我有这样的东西:
projectFiles1 = glob.glob( os.path.join(projectDir, '*.txt') )
projectFiles2 = glob.glob( os.path.join(projectDir, '*.mdown') )
projectFiles3 = glob.glob( os.path.join(projectDir, '*.markdown') )
当前回答
最简单的方法是使用itertools.chain
from pathlib import Path
import itertools
cwd = Path.cwd()
for file in itertools.chain(
cwd.rglob("*.txt"),
cwd.rglob("*.md"),
):
print(file.name)
其他回答
这应该有用:
import glob
extensions = ('*.txt', '*.mdown', '*.markdown')
for i in extensions:
for files in glob.glob(i):
print (files)
使用扩展列表并遍历
from os.path import join
from glob import glob
files = []
extensions = ['*.gif', '*.png', '*.jpg']
for ext in extensions:
files.extend(glob(join("path/to/dir", ext)))
print(files)
Glob返回一个列表:为什么不只是多次运行它并连接结果呢?
from glob import glob
project_files = glob('*.txt') + glob('*.mdown') + glob('*.markdown')
也许有更好的办法，但是:
import glob
types = ('*.pdf', '*.cpp') # the tuple of file types
files_grabbed = []
for files in types:
files_grabbed.extend(glob.glob(files))
# files_grabbed is the list of pdf and cpp files
也许还有其他的方法，所以等待别人提出更好的答案。
最简单的方法是使用itertools.chain
from pathlib import Path
import itertools
cwd = Path.cwd()
for file in itertools.chain(
cwd.rglob("*.txt"),
cwd.rglob("*.md"),
):
print(file.name)
