当前位置: 首页 > 图灵资讯 > 行业资讯> python如何读取文件名

python如何读取文件名

来源:图灵python
时间: 2024-07-30 09:56:40

本文教程操作环境:windows7系统Python 3.9.1,DELL G3电脑。

1、os.listdir()函数只获得当前路径下的文件名,不包括子目录中的文件,需要通过递归获得的文件名。

#在指定路径下返回所有文件,不包括子目录。
path1=r"C:\Users\11764\Desktop\Data"
f=os.listdir(path1)
print(f)


output:

['2020-09-16',
'2020-10-11',
'baidu_index_0625.xlsx',
'city_id.xlsx',
'city_index_0625.xlsx',
'province_id.xlsx',
'province_index_0625.xlsx']

2、os.path.splitext()函数将路径分为文件名和扩展名(后缀)。

os.path.splitext(svm.model) 结果是svmm[0] 以上是对所有文件名称的判断,然后各取所需,添加到相应的列表中。

importos
#保存不同模型的目录名(绝对路径)
file_dir=r'C:\Users\zhangyh4Desktopxietong1012model'
bayes_list=[]#贝叶斯模型列表
svm_list=[]#svm模型列表
xgboost_list=[]#xgbost模型列表
#root是指当前目录路径(文件夹的绝对路径)
#dirs是指路径下的所有子目录(文件夹中的文件夹)
#files是指路径下的所有文件(文件夹中的所有文件)
forroot,dirs,filesinos.walk(file_dir):
forfileinfiles:
ifos.path.splitext(file)[0]='bayes':
bayes_list.append(os.path.join(root,file))
elifos.path.splitext(file)[0]='svm':
svm_list.append(os.path.join(root,file))
elifos.path.splitext(file)[0]='xgboost':
xgboost_list.append(os.path.join(root,file))

以上是python读取文件名的方法。我们可以通过os模块下的函数来解决这个问题,并获得不同类型的文件名。我们也可以在学习后进行这个测试。更多Python学习指导:python基础教程