python如何获取文件名

162次观看

标签：文件名 python

以下回答来自环球青藤，点击开启技能学习>

老师回答

张老师 Python编程

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

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

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

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

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

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

import os
# 保存不同模型的目录名(绝对路径)
file_dir = r'C:Userszhangyh4Desktopxietong1012model'
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中如何进行集合交集运算？

刘老师 Python编程

方法一：使用&计算

lis1,lis2=[1,2],[1]

print(list(set(lis1) & set(lis2)))

输出为：[1]

方法二：使用python推导式计算

lst_a = [1,2,3,4,5]
lst_b = [3,4,5,6,7]
lst_c = [x for x in lst_b if x in lst_a]
# lst_c = [x for x in lst_a if x in lst_b]
print(lst_c)

输出

[3, 4, 5]

方法三：使用python集合的内置方法计算

L = [[1,2,3,4], [2,3,4,5], [3,4,5,6],[1,2,3,4,5,6]]
P = set(L[0]).intersection(*L[1:])
print(P)

输出

{3, 4}

python中字典如何按照value排序？

付老师 Python编程

方法一：key使用lambda匿名函数取value进行排序

dict= {'a':1,'b':4,'c':2}
sorted(dict.items(),key = lambda x:x[1],reverse = True)

方法二：使用operator的itemgetter进行排序

test_data_6=sorted(dict_data.items(),key=operator.itemgetter(1))
test_data_7=sorted(dict_data.items(),key=operator.itemgetter(1),reverse=True)
print(test_data_6) #[(8, 2), (10, 5), (7, 6), (6, 9), (3, 11)]
print(test_data_7) #[(3, 11), (6, 9), (7, 6), (10, 5), (8, 2)]

方法三：key和value分装成元祖，再进行排序

f = zip(d.keys(), d.values())
c = sorted(f)
print(c)

用Python编程需要哪些软件?

付老师 Python编程

一、终端：Upterm

Upterm简略好用，它是一个全平台的终端，能够说是终端里的IDE，有着强大的自动补全功用。

二、交互式解说器：PtPython

一个交互式的Python解说器。支撑语法高亮、提示乃至是vim和emacs的键入形式。其实我们在课程里供给的在线终端也内置了ptPython。

三、包管理：Anaconda

能帮你装置好许多麻烦的软件，包括：Python环境、pip包管理工具、常用的库、装备好环境路径等等。用Python搞数据方面的事情，就装置Anaconda就好了，它乃至开发了一套JIT的解说器Numba。所以Anaconda有了JIT之后，对线上科学计算效率要求比较高的东西也能够搞了。

四、脚本引擎：QPython

QPython是一个能够在安卓体系上运转Python脚本引擎，整合了Python解说器、Console、编辑器和SL4A库，在安卓设备上你照样能够玩转Python。