python读取文件名称生成list的方法


Posted in Python onApril 27, 2018

经常需要读取某个文件夹下所有的图像文件。

我使用python写了个简单的代码,读取某个文件夹下某个后缀的文件,将文件名生成为文本(csv格式)

import fnmatch
import os
import pandas as pd
import numpy as np 
import sys

InputStra = sys.argv[1]
InputStrb = sys.argv[2]

def ReadSaveAddr(Stra,Strb):
 #print(Stra)
 #print(Strb)
 print("Read :",Stra,Strb)
 a_list = fnmatch.filter(os.listdir(Stra),Strb)
 print("Find = ",len(a_list))
 df = pd.DataFrame(np.arange(len(a_list)).reshape((len(a_list),1)),columns=['Addr']) 
 df.Addr = a_list
 #print(df.head())
 df.to_csv('Get.lst',columns=['Addr'],index=False,header=False)
 print("Write To Get.lst !")

ReadSaveAddr(InputStra,InputStrb)

上面代码保存为:GetLst.py

使用时:

在cmd窗口输入:

python GetLst.py F:/train/pos *.png

发现上面代码不能深入到下一层目录,又做了点修改:

def ReadSaveAddr2(Stra,Strb):
 df = pd.DataFrame(np.arange(0).reshape(0,1),columns=['Addr']) 
 print(df)
 path = InputStra
 for dirpath,dirnames,filenames in os.walk(path):
  #for filename in filenames:
  a_list = fnmatch.filter(os.listdir(dirpath),Strb)
  if len(a_list):
   dft = pd.DataFrame(np.arange(len(a_list)).reshape((len(a_list),1)),columns=['Addr']) 
   dft.Addr = a_list
   dft.Addr = dirpath + '\\' + dft.Addr#输出绝对路径
   frames = [df,dft]
   df = pd.concat(frames)
   print(df.shape)
 df.to_csv('Get.lst',columns='Addr'],index=False,header=False)
 print("Write To Get.lst !")

以上这篇python读取文件名称生成list的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中使用异常处理来判断运行的操作系统平台方法
Jan 22 Python
Python实现Windows和Linux之间互相传输文件(文件夹)的方法
May 08 Python
python编程实现12306的一个小爬虫实例
Dec 27 Python
python编程培训 python培训靠谱吗
Jan 17 Python
python3对拉勾数据进行可视化分析的方法详解
Apr 03 Python
Python中单线程、多线程和多进程的效率对比实验实例
May 14 Python
Python学习笔记之Django创建第一个数据库模型的方法
Aug 07 Python
python利用openpyxl拆分多个工作表的工作簿的方法
Sep 27 Python
Pycharm内置终端及远程SSH工具的使用教程图文详解
Mar 19 Python
python如何导出微信公众号文章方法详解
Aug 31 Python
python PIL模块的基本使用
Sep 29 Python
python 学习GCN图卷积神经网络
May 11 Python
python实现图书馆研习室自动预约功能
Apr 27 #Python
Python实现去除列表中重复元素的方法小结【4种方法】
Apr 27 #Python
详解python多线程、锁、event事件机制的简单使用
Apr 27 #Python
Python Requests模拟登录实现图书馆座位自动预约
Apr 27 #Python
Python多线程中阻塞(join)与锁(Lock)使用误区解析
Apr 27 #Python
python队列queue模块详解
Apr 27 #Python
浅谈tensorflow1.0 池化层(pooling)和全连接层(dense)
Apr 27 #Python
You might like
php session_start()关于Cannot send session cache limiter - headers already sent错误解决方法
2009/11/27 PHP
PHP下常用正则表达式整理
2010/10/26 PHP
PHP不用递归遍历目录下所有文件的代码
2014/07/04 PHP
PHP+MySQL实现的简单投票系统实例
2016/02/24 PHP
PHP判断用户是否已经登录(跳转到不同页面或者执行不同动作)
2016/09/22 PHP
PHP实现Huffman编码/解码的示例代码
2018/04/20 PHP
利用PHP计算有多少小于当前数字的数字方法示例
2020/08/26 PHP
jquery教程ajax请求json数据示例
2014/01/13 Javascript
javascript匿名函数应用示例介绍
2014/03/07 Javascript
js动态添加onclick事件可传参数与不传参数
2014/07/29 Javascript
JS中处理时间之setUTCMinutes()方法的使用
2015/06/12 Javascript
如何防止INPUT按回车自动提交表单FORM
2016/12/06 Javascript
canvas实现流星雨的背景效果
2017/01/13 Javascript
浅谈vue.js中v-for循环渲染
2017/07/26 Javascript
使用JavaScript进行表单校验功能
2017/08/01 Javascript
vue-baidu-map 进入页面自动定位的解决方案(推荐)
2018/04/28 Javascript
详解angular脏检查原理及伪代码实现
2018/06/08 Javascript
Nodejs实现爬虫抓取数据实例解析
2018/07/05 NodeJs
React Hooks的深入理解与使用
2018/11/12 Javascript
Python将图片批量从png格式转换至WebP格式
2020/08/22 Python
浅谈Python2、Python3相对路径、绝对路径导入方法
2018/06/22 Python
python绘制直方图和密度图的实例
2019/07/08 Python
Pycharm2020.1安装无法启动问题即设置中文插件的方法
2020/08/07 Python
教师年度考核自我鉴定
2014/01/19 职场文书
计算机专业优秀大学生自我总结
2014/01/21 职场文书
幼儿教师工作感言
2014/02/14 职场文书
房地产项目建议书
2014/03/12 职场文书
清正廉洁演讲稿
2014/05/22 职场文书
文秘应届生求职信
2014/07/05 职场文书
个人整改方案范文
2014/10/25 职场文书
社区节水倡议书
2015/04/29 职场文书
工作年限证明模板
2015/06/15 职场文书
详解CocosCreator消息分发机制
2021/04/16 Javascript
学会用Python实现滑雪小游戏,再也不用去北海道啦
2021/05/20 Python
Django REST framework 限流功能的使用
2021/06/24 Python
frg-100简单操作(设置)说明
2022/04/05 无线电