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线程锁(thread)学习示例
Dec 04 Python
Python脚本文件打包成可执行文件的方法
Jun 02 Python
浅谈python字典多键值及重复键值的使用
Nov 04 Python
python在非root权限下的安装方法
Jan 23 Python
cmd运行python文件时对结果进行保存的方法
May 16 Python
python 3.6.2 安装配置方法图文教程
Sep 18 Python
Python实现分段线性插值
Dec 17 Python
Django中的cookie和session
Aug 27 Python
Python装饰器用法与知识点小结
Mar 09 Python
python 6行代码制作月历生成器
Sep 18 Python
Python之多进程与多线程的使用
Feb 23 Python
Python通过m3u8文件下载合并ts视频的操作
Apr 16 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中几种常见的超时处理全面总结
2012/09/11 PHP
php绘制一条直线的方法
2015/01/24 PHP
PHP实现的蚂蚁爬杆路径算法代码
2015/12/03 PHP
MAC下通过改apache配置文件切换php多版本的方法
2017/04/26 PHP
PHP htmlspecialchars_decode()函数用法讲解
2019/03/01 PHP
laravel 获取当前url的别名方法
2019/10/11 PHP
javascript 面向对象继承
2009/11/26 Javascript
jQuery动态添加 input type=file的实现代码
2012/06/14 Javascript
js 遍历json返回的map内容示例代码
2013/10/29 Javascript
什么是 AngularJS?AngularJS简介
2014/12/06 Javascript
jQuery控制文本框只能输入数字和字母及使用方法
2016/05/26 Javascript
jQuery ajax全局函数处理session过期后的ajax跳转问题
2016/06/03 Javascript
JavaScript中Number对象的toFixed() 方法详解
2016/09/02 Javascript
AngularJS 自定义过滤器详解及实例代码
2016/09/14 Javascript
代码详解JS操作剪贴板
2018/02/11 Javascript
vue.js实现的经典计算器/科学计算器功能示例
2018/07/11 Javascript
详解用vue2.x版本+adminLTE开源框架搭建后台应用模版
2019/03/15 Javascript
JS+CSS3实现的简易钟表效果示例
2019/04/13 Javascript
微信小程序收货地址API兼容低版本解决方法
2019/05/18 Javascript
layui 对弹窗 form表单赋值的实现方法
2019/09/04 Javascript
微信小程序canvas分享海报功能
2019/10/31 Javascript
微信小程序点击滚动到指定位置的实现
2020/05/22 Javascript
Python实现抓取页面上链接的简单爬虫分享
2015/01/21 Python
python友情链接检查方法
2015/07/08 Python
python itchat实现微信自动回复的示例代码
2017/08/14 Python
Python使用sklearn库实现的各种分类算法简单应用小结
2019/07/04 Python
django 多对多表的创建和插入代码实现
2019/09/09 Python
python 代码运行时间获取方式详解
2020/09/18 Python
CSS3绘制圆角矩形的简单示例
2015/09/28 HTML / CSS
HTML5印章绘制电子签章图片(中文英文椭圆章、中文英文椭圆印章)
2019/06/03 HTML / CSS
Lookfantastic德国官网:英国知名美妆购物网站
2017/06/11 全球购物
英国最受欢迎的在线隐形眼镜商店:VisionDirect.co.uk
2018/12/06 全球购物
高三毕业生自我鉴定
2013/12/20 职场文书
房屋产权共有协议书范本
2014/11/03 职场文书
大学生求职简历自我评价
2015/03/02 职场文书
一文解答什么是MySQL的回表
2022/08/05 MySQL