python读取文件名称生成list的方法
作者：masbbx123 发布时间：2021-12-21 02:40:34
标签：python,读取,文件,list
经常需要读取某个文件夹下所有的图像文件。
我使用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 !")
来源：https://blog.csdn.net/masbbx123/article/details/78954294
0
投稿猜你喜欢
- <?php list($usec, $sec) = explode(' ', microtime()); $timer
概述从今天开始, 小白我将带领大家一起来补充一下 数据库的知识.条件查询我们可以使用关键词Where来指定条件, 用于插入, 修改删除或者查- 1.文件写入#打开文件，路径不对会报错f = open(r"C:\Users\jm\Desktop\pyfile.txt"
- 很多时候关心的是优化SELECT 查询，因为它们是最常用的查询，而且确定怎样优化它们并不总是直截了当。相对来说，将数据装入数据库是直截了当的
- 在线阅读与印刷品阅读有着很大的不同，原因之一是屏幕会导致注意力问题。文字很难成为重点，而闪烁的横幅、明亮的图片又更加会转移对实际的博客内容的
- 本文实例讲述了Python可变参数用法。分享给大家供大家参考，具体如下：#!/usr/bin/env python# -*- coding:
- 前言在web应用中经常会遇到数据验证问题，普通的验证方法比较繁琐，这里介绍一个使用比较多的包validator。原理将验证规则写在struc
- 前言最近在研究怎么对图片资源进行无损压缩，网上也找了一些资料。总而言之，收获不少，所以想对最近的学习做个总结。无损压缩其实是相对而言的，目的
- 如下所示：#encoding=utf-8print '中国'#字典的一键多值print'方案一 list作为dict
- 安装完 anaconda运行如下代码执行不了import numpy as npimport os,sys#获取当前文件夹，并根据文件名de
- 智能聊天一、 概述我们将我们的qq聊天机器人的环境配置好后，其就可以开始接收消息啦！那么，我们除了可以接收特定的消息，是不是还需要接收那些不
- SQL的扩展的删除与恢复 删除 代码如下:use master exec spdropextendedproc “xpcmdshell“ e
- 前言要在pandas.DataFrame中的任何位置检索或更改数据，可以使用at，iat，loc，iloc。位置的指定方法at，loc：行标
- 这里是一个使用日期函数的例子。下面的查询选择了所有记录，其date_col的值是在最后30天以内： mysql>
- 环境准备python3.5.4windows redispip install celerypip install rediswindows
- 前言：Pandas 中应用 query 函数来进行数据筛选。query 函数的一般用法如下：df.query('expression
- 可能是IP没设置好问题：MySQL权限设置正确，但仍无法远程访问。通过telnet发现3306端口未打开。分析：MySQL默认只绑定127.
- type()动态语言和静态语言最大的不同，就是函数和类的定义，不是编译时定义的，而是运行时动态创建的。比方说我们要定义一个Hello的cla
- 本文研究的主要是python+matplotlib实现动态绘制图片（交互式绘图）的相关内容，具体介绍和实现代码如下所示。最近在研究动态障碍物
- 之前一直使用的是MySQL5.7，但由于MySQL增加了一些新特性，所以选择了更新。下载MySQL进入MySQL官网下载地址，选择Windo
