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中文乱码的解决方法
Nov 04 Python
python计算最小优先级队列代码分享
Dec 18 Python
Python中多线程thread与threading的实现方法
Aug 18 Python
Python入门之三角函数atan2()函数详解
Nov 08 Python
python之cv2与图像的载入、显示和保存实例
Dec 05 Python
在PyCharm的 Terminal(终端)切换Python版本的方法
Aug 02 Python
Django通过dwebsocket实现websocket的例子
Nov 15 Python
Python实现手机号自动判断男女性别(实例解析)
Dec 22 Python
基于Python数据结构之递归与回溯搜索
Feb 26 Python
20行Python代码实现一款永久免费PDF编辑工具的实现
Aug 27 Python
Python使用windows设置定时执行脚本
Nov 12 Python
python工具——Mimesis的简单使用教程
Jan 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
thinkPHP的Html模板标签使用方法
2012/11/13 PHP
js trim函数 去空格函数与正则集锦
2009/11/20 Javascript
js setattribute批量设置css样式
2009/11/26 Javascript
javascript 全选与全取消功能的实现代码
2012/12/23 Javascript
原生JS实现表单checkbook获取已选择的值
2013/07/21 Javascript
js 中的switch表达式使用示例
2020/06/03 Javascript
javascript eval(func())使用示例
2013/12/05 Javascript
js采用map取到id集合组并且实现点击一行选中一行
2013/12/16 Javascript
可插入图片的TEXT文本框
2013/12/27 Javascript
Jquery Post处理后不进入回调的原因及解决方法
2014/07/15 Javascript
jQuery插件windowScroll实现单屏滚动特效
2015/07/14 Javascript
微信小程序 前端源码逻辑和工作流详解
2016/10/08 Javascript
Bootstrap中datetimepicker使用小结
2016/12/28 Javascript
浅谈js中startsWith 函数不能在任何浏览器兼容的问题
2017/03/01 Javascript
Vue.js结合Ueditor富文本编辑器的实例代码
2017/07/11 Javascript
详解tween.js 中文使用指南
2018/01/05 Javascript
Koa 中的错误处理解析
2019/04/09 Javascript
关于微信小程序map组件z-index的层级问题分析
2019/07/09 Javascript
微信小程序封装分享与分销功能过程解析
2019/08/13 Javascript
微信小程序全局变量的设置、使用、修改过程解析
2019/09/24 Javascript
微信小程序后端实现授权登录
2020/02/24 Javascript
Flask入门教程实例:搭建一个静态博客
2015/03/27 Python
python数组过滤实现方法
2015/07/27 Python
python实现简单神经网络算法
2018/03/10 Python
Python浅复制中对象生存周期实例分析
2018/04/02 Python
Python3数字求和的实例
2019/02/19 Python
解决运行出现'dict' object has no attribute 'has_key'问题
2020/07/15 Python
入股协议书
2014/04/14 职场文书
协议书怎么写
2014/04/21 职场文书
农民工讨薪标语
2014/06/26 职场文书
购房意向书
2014/08/30 职场文书
2014年教师党员自我评议
2014/09/19 职场文书
迁户口计划生育证明
2014/10/19 职场文书
导游词400字
2015/02/13 职场文书
酒桌上的开场白
2015/06/01 职场文书
golang 语言中错误处理机制
2021/08/30 Golang