有的时候,我们需要读取图片名,写入表格中,以便结合图片的其他信息,做进一步的分析。 假如,现在要读取存放在E盘的origin_file文件夹,读取里面的图片名将其写入到excel文件img.xlsx中。 首先,需要读取图片文件夹路径
import pandas as pd
import os
os.chdir('E:\\')
path='origin_file'
然后,pandas建立空白excel文件“img.xlsx”
writer=pd.ExcelWriter("img.xlsx")
然后,遍历图片文件夹里的文件,并将文件名写入新的列表
for root,dirs,files in os.walk(path):
list=[]
for file in files:
file=file.rstrip(".jpg")
list.append(file)
然后,将列表list嵌套进字典data中,转换成dataframe格式储存在开始建立的excel中
dict_={'filename':list}
df=pd.DataFrame(dict_)
df.to_excel(writer,'sheet1',startcol=0,index=False )
writer.save()
文件夹里的图片的图片名就被写入excel表格中了 思路总结: 读取目标文件夹——建立新excel——图片名写入列表——列表嵌套字典转成dataframe格式——保存在excel 完整代码如下:
import pandas as pd
import os
os.chdir('E:\\')
path='origin_file'
writer=pd.ExcelWriter("img.xlsx")
for root,dirs,files in os.walk(path):
list=[]
for file in files:
file=file.rstrip(".jpg")
list.append(file)
dict_={'filename':list}
df=pd.DataFrame(dict_)
df.to_excel(writer,'sheet1',startcol=0,index=False )
writer.save()
创作不易,请大家多多点赞,收藏,支持一下!
|