python目录,文件相关操作

python目录,文件相关操作

  • 一、获取路径
  • 二、筛选文件名
  • 三、判断文件/目录
  • 四、获取文件名

训练模型时经常需要对数据文件进行相关处理,这里实时进行总结

一、获取路径

函数 作用
os.path.abspath(__ file__) 返回当前文件的绝对路径,也可是指定路径的绝对路径
os.path.dirname(path) 返回路径的上一级路径
os.path.join() 可以进行路径拼接
os.listdir(path) 得到该路径下的所有文件名(注意仅仅是当前目录下)

二、筛选文件名

函数 作用
filter() 筛选满足条件的文件名

例子(从下面数列中选取 .jpg图片):

img_names=['alexnet-owt-4df8aa71.pth', 'Golden Retriever from baidu.jpg', 
           'imagenet1000.json', 'imagenet_classnames.txt', 'test', 
           'tiger cat.jpg', 'train']
img_names = list(filter(lambda x:x.endswith('.jpg'),img_names))
print(iimg_names)

结果:
>>['Golden Retriever from baidu.jpg', 'tiger cat.jpg']

三、判断文件/目录

函数 作用
os.path.exists(path) 判断路径是否存在
os.path.isdir(path) 判断是否是文件夹
os.path.isfile(path) 判断是否是文件

四、获取文件名

函数 作用
os.path.basename(path) 获取目录的最后文件名

你可能感兴趣的:(神经网络基础及相关处理,深度学习,人工智能)