有没有更好的方法来使用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') )
当前回答
从前面的答案
glob('*.jpg') + glob('*.png')
这是一个较短的问题，
from glob import glob
extensions = ['jpg', 'png'] # to find these filename extensions
# Method 1: loop one by one and extend to the output list
output = []
[output.extend(glob(f'*.{name}')) for name in extensions]
print(output)
# Method 2: even shorter
# loop filename extension to glob() it and flatten it to a list
output = [p for p2 in [glob(f'*.{name}') for name in extensions] for p in p2]
print(output)
其他回答
使用扩展列表并遍历
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)
最简单的方法是使用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
import pandas as pd
df1 = pd.DataFrame(columns=['A'])
for i in glob.glob('C:\dir\path\*.txt'):
df1 = df1.append({'A': i}, ignore_index=True)
for i in glob.glob('C:\dir\path\*.mdown'):
df1 = df1.append({'A': i}, ignore_index=True)
for i in glob.glob('C:\dir\path\*.markdown):
df1 = df1.append({'A': i}, ignore_index=True)
如果你使用pathlib，试试这个:
import pathlib
extensions = ['.py', '.txt']
root_dir = './test/'
files = filter(lambda p: p.suffix in extensions, pathlib.Path(root_dir).glob('**/*'))
print(list(files))
一句话:
img_ext = (".jpg"， ".jpeg"， ".jpeg"， "。就是“,”。jfi”、“.jif”、“jpg”)
目录= './'
files = [glob.glob(目录+'/*')中的文件的文件if file.endswith(img_ext)]
