有没有更好的方法来使用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') )
当前回答
import os
import glob
projectFiles = [i for i in glob.glob(os.path.join(projectDir,"*")) if os.path.splitext(i)[-1].lower() in ['.txt','.markdown','.mdown']]
Os.path.splitext将返回filename & .extension
filename, .extension = os.path.splitext('filename.extension')
.lower()将字符串转换为小写
其他回答
虽然Python的默认glob并没有真正遵循Bash的glob，但您可以使用其他库来做到这一点。我们可以在wcmatch的glob中启用大括号。
>>> from wcmatch import glob
>>> glob.glob('*.{md,ini}', flags=glob.BRACE)
['LICENSE.md', 'README.md', 'tox.ini']
你甚至可以使用扩展glob模式，如果这是你的偏好:
from wcmatch import glob
>>> glob.glob('*.@(md|ini)', flags=glob.EXTGLOB)
['LICENSE.md', 'README.md', 'tox.ini']
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)
不是glob，这里是另一种使用列表理解的方式:
extensions = 'txt mdown markdown'.split()
projectFiles = [f for f in os.listdir(projectDir)
if os.path.splitext(f)[1][1:] in extensions]
这对我很管用!
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
你可以用这个:
project_files = []
file_extensions = ['txt','mdown','markdown']
for file_extension in file_extensions:
project_files.extend(glob.glob(projectDir + '*.' + file_extension))
