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 相关文章推荐
python3图片转换二进制存入mysql
Dec 06 Python
python检测lvs real server状态
Jan 22 Python
python的类变量和成员变量用法实例教程
Aug 25 Python
Python解析网页源代码中的115网盘链接实例
Sep 30 Python
Python中用于返回绝对值的abs()方法
May 14 Python
TensorFlow实现RNN循环神经网络
Feb 28 Python
Python之csv文件从MySQL数据库导入导出的方法
Jun 21 Python
python+opencv实现车牌定位功能(实例代码)
Dec 24 Python
浅谈python多线程和多线程变量共享问题介绍
Apr 17 Python
Python 多线程C段扫描、检测 Ping扫描脚本的实现
Sep 03 Python
Python self用法详解
Nov 28 Python
实战Python爬虫爬取酷我音乐
Apr 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编实现程动态图像的创建代码
2008/09/28 PHP
获取URL文件名后缀
2013/10/24 PHP
ThinkPhP+Apache+PHPstorm整合框架流程图解
2020/11/23 PHP
jquery编写日期选择器
2017/03/16 Javascript
微信小程序返回多级页面的实现方法
2017/10/27 Javascript
JavaScript 日期时间选择器一些小结
2018/04/02 Javascript
AngularJS使用$http配置对象方式与服务端交互方法
2018/08/13 Javascript
写gulp遇到的ES6问题详解
2018/12/03 Javascript
JS/jQuery实现超简单的Table表格添加,删除行功能示例
2019/07/31 jQuery
vue.js基于v-for实现批量渲染 Json数组对象列表数据示例
2019/08/03 Javascript
vue 源码解析之虚拟Dom-render
2019/08/26 Javascript
ES10的13个新特性示例(小结)
2019/09/23 Javascript
vue 解决移动端弹出键盘导致页面fixed布局错乱的问题
2019/11/06 Javascript
vue不操作dom实现图片轮播的示例代码
2019/12/18 Javascript
js实现带搜索功能的下拉框
2020/01/11 Javascript
vue 使用 vue-pdf 实现pdf在线预览的示例代码
2020/04/26 Javascript
[03:36]2014DOTA2 TI小组赛综述 八强诞生进军钥匙球馆
2014/07/15 DOTA
Python实现递归遍历文件夹并删除文件
2016/04/18 Python
python实现海螺图片的方法示例
2019/05/12 Python
python selenium 执行完毕关闭chromedriver进程示例
2019/11/15 Python
python绘制彩虹图
2019/12/16 Python
Python Numpy,mask图像的生成详解
2020/02/19 Python
Funko官方商店:源自美国,畅销全球搪胶收藏玩偶
2018/09/15 全球购物
亚马逊新加坡官方网站:Amazon.sg
2020/03/25 全球购物
司机辞职报告范文
2014/01/20 职场文书
全民健身日活动方案
2014/01/29 职场文书
简单的离婚协议书范本
2014/11/16 职场文书
2015年学校减负工作总结
2015/05/19 职场文书
小王子读书笔记
2015/06/29 职场文书
2015年校本培训工作总结
2015/07/24 职场文书
新郎新娘致辞
2015/07/31 职场文书
初一军训感言
2015/08/01 职场文书
商业计划书如何写?关键问题有哪些?
2019/07/11 职场文书
2019年国庆祝福语(70句)
2019/09/19 职场文书
html+css实现滚动到元素位置显示加载动画效果
2021/08/02 HTML / CSS
MySQL 如何限制一张表的记录数
2021/09/14 MySQL