python培训之如何获取文件名

发布时间:2022-08-04 09:43:12 人气:50 作者:多测师

  python获取文件名的方法:

  在做项目中我们会有这个需求的,比如你训练数据过程也许会保存很多的模型,这个时候你要读取这些模型,去对测试数据进行预测,那

  么你就要获取目录下的这些模型的文件名,从而读取这些模型。

  方法有很多,但我还是只说一种——os.walk的方法

  在这里,我再加一个特殊的需求,比如你目录下有三种模型,那么你要把这三种模型的所有文件名加入到不同的列表中,怎么办?——简

  而言之,就是取出目录下的特定文件

  我的model目录下有三种不同的模型,我要取出它们,分别放到不同的列表中,注意看我的注释

  import os

python培训之如何获取文件名

  # 保存不同模型的目录名(绝对路径)

  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))

  以上内容为大家介绍了python培训之如何获取文件名,希望对大家有所帮助,如果想要了解更多Python相关知识,请关注多测师。https://www.e70w.com/xwzx/


返回列表
在线客服
联系方式

热线电话

17727591462

上班时间

周一到周五

二维码
线