不懂python获取文件名的方法？其实想解决这个问题也不难，下面让小编带着大家一起学习怎么去解决，希望大家阅读完这篇文章后大所收获。
在做项目中我们会有这个需求的，比如你训练数据过程也许会保存很多的模型，这个时候你要读取这些模型，去对测试数据进行预测，那
么你就要获取目录下的这些模型的文件名，从而读取这些模型。
方法有很多，但我还是只说一种——os.walk的方法
在这里，我再加一个特殊的需求，比如你目录下有三种模型，那么你要把这三种模型的所有文件名加入到不同的列表中，怎么办？——简
而言之，就是取出目录下的特定文件
我的model目录下有三种不同的模型，我要取出它们，分别放到不同的列表中，注意看我的注释
import os # 保存不同模型的目录名(绝对路径) file_dir = r'C:\Users\zhangyh5\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))
感谢你能够认真阅读完这篇文章，希望小编分享python获取文件名的方法内容对大家有帮助，同时也希望大家多多支持亿速云，关注亿速云行业资讯频道，遇到问题就找亿速云，详细的解决方法等着你来学习!
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
