有没有更好的方法来使用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') )
当前回答
from glob import glob
files = glob('*.gif')
files.extend(glob('*.png'))
files.extend(glob('*.jpg'))
print(files)
如果你需要指定一个路径，循环匹配模式，并保持连接在循环中简单:
from os.path import join
from glob import glob
files = []
for ext in ('*.gif', '*.png', '*.jpg'):
files.extend(glob(join("path/to/dir", ext)))
print(files)
其他回答
不是glob，这里是另一种使用列表理解的方式:
extensions = 'txt mdown markdown'.split()
projectFiles = [f for f in os.listdir(projectDir)
if os.path.splitext(f)[1][1:] in extensions]
一句俏皮话，只是为了好玩。
folder = "C:\\multi_pattern_glob_one_liner"
files = [item for sublist in [glob.glob(folder + ext) for ext in ["/*.txt", "/*.bat"]] for item in sublist]
输出:
['C:\\multi_pattern_glob_one_liner\\dummy_txt.txt', 'C:\\multi_pattern_glob_one_liner\\dummy_bat.bat']
与@BPL相同的答案(计算效率高)，但它可以处理任何glob模式，而不是扩展:
import os
from fnmatch import fnmatch
folder = "path/to/folder/"
patterns = ("*.txt", "*.md", "*.markdown")
files = [f.path for f in os.scandir(folder) if any(fnmatch(f, p) for p in patterns)]
这种解决方案既高效又方便。它还与glob的行为紧密匹配(请参阅文档)。
注意，使用内置包pathlib会更简单:
from pathlib import Path
folder = Path("/path/to/folder")
patterns = ("*.txt", "*.md", "*.markdown")
files = [f for f in folder.iterdir() if any(f.match(p) for p in patterns)]
这对我很管用!
split('.')[-1]
上面的代码分开了文件名后缀(*.xxx)，所以它可以帮助你
for filename in glob.glob(folder + '*.*'):
print(folder+filename)
if filename.split('.')[-1] != 'tif' and \
filename.split('.')[-1] != 'tiff' and \
filename.split('.')[-1] != 'bmp' and \
filename.split('.')[-1] != 'jpg' and \
filename.split('.')[-1] != 'jpeg' and \
filename.split('.')[-1] != 'png':
continue
# Your code
from glob import glob
files = glob('*.gif')
files.extend(glob('*.png'))
files.extend(glob('*.jpg'))
print(files)
如果你需要指定一个路径，循环匹配模式，并保持连接在循环中简单:
from os.path import join
from glob import glob
files = []
for ext in ('*.gif', '*.png', '*.jpg'):
files.extend(glob(join("path/to/dir", ext)))
print(files)
