python 获取文件夹下文件的名字

前言

    承接上一篇文章.上一篇文章.利用glob.glob()可以获取某一目录下所有符合指定条件的文件的路径,并以列表返回.那么如果说文件的名字就是我们在做文本分类任务时的类别呢,那又该如何快速获取文件的名字呢?

1.python获取文件名字

    使用os模块:

import os
data_path = "/home/fml/桌面/pytorch学习/pytorch基础/data/names/German.txt"
basename = os.path.basename(data_path)
print(basename)
category = os.path.splitext(basename)
print(category)

    先看执行结果:

German.txt
('German', '.txt')

    然后使用category[0]就获取到文件的名字了.
    解释:

• os.path.basename(data_path):返回data_path最后的文件名
• os.path.splitext(basename):将文件名和扩展名分开.

写在最后

    本文是个人的一些学习笔记,如有侵权,请及时联系我进行删除,谢谢大家.