导语：
本文主要介绍了关于python如何获取文件名的相关知识，包括python获取文件夹名称，以及Python右键获取文件名这些编程知识，希望对大家有参考作用。
我们在项目中会有这个要求。例如，你可能在训练数据的过程中保存了很多模型。这时候就需要阅读这些模型来预测测试数据。然后
然后你需要在目录中获取这些模型的文件名来读取这些模型。
方法有很多，但我还是只说一种——os.walk的方法
在这里，我添加一个特殊要求。例如，如果你的目录中有三个模型，那么你希望将这三个模型的所有文件名添加到不同的列表中。你该怎么办？ - 简单的
而言之，就是取出目录下的特定文件
我的model目录里面有3个不同的model，我想把它们拿出来放到不同的list里，注意我的评论
import os
# 保存不同模型的目录名(绝对路径)
file_dir = r'C:\Users\zhangyh4\Desktop\xietong1012\model'
bayes_list = [] # 贝叶斯模型列表
svm_list = [] # svm模型列表
xgboost_list = [] # xgboost模型列表
# root是指当前目录路径(文件夹的绝对路径)
# dirs是指路径下所有的子目录(文件夹里的文件夹)
# files是指路径下所有的文件(文件夹里所有的文件)
for root,dirs,files in os.walk(file_dir):
for file in files:
if os.path.splitext(file)[0] == 'bayes':
bayes_list.append(os.path.join(root,file))
elif os.path.splitext(file)[0] == 'svm':
svm_list.append(os.path.join(root,file))
elif os.path.splitext(file)[0] == 'xgboost':
xgboost_list.append(os.path.join(root,file))
推荐学习《
》
本文为原创文章，版权归知行编程网所有，欢迎分享本文，转载请保留出处！
内容反馈