网站建设资讯

NEWS

网站建设资讯

python读取文件名称生成list的方法-创新互联

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

成都创新互联公司主营剑河网站建设的网络公司,主营网站建设方案,重庆APP软件开发,剑河h5成都微信小程序搭建,剑河网站营销推广欢迎剑河等地区企业咨询

我使用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)

本文题目:python读取文件名称生成list的方法-创新互联
文章起源:http://njwzjz.com/article/cophhi.html