读取文件名写入csv

首先读取文件夹内的文件名到list中，然后转换成narray，写入dataframe

（这里将png后缀的文件改写后缀和地址，是为了指向另一个文件夹内的，这是特殊的需要，如果不需要的话可以不写）

然后将dataframe写入csv

其中insert是增加列的方法

import pandas as pd
import os
import numpy as np

save_name = 't.csv'
jpgfilelist = os.listdir(r'C:\Users\mage\Desktop\output')
jpgfilename = np.array(['C:/Users/mage/Desktop/output/' + file for file in jpgfilelist if file.endswith('.jpg')],
                       dtype=object)

dataframe = pd.DataFrame(jpgfilename.reshape(len(jpgfilename), 1))
# dataframe还有另一种写入方法
# dataframe.insert(dataframe.shape[1],'1',pngfilename.reshape(len(pngfilename), 1))

dataframe.to_csv(save_name, encoding='utf-8', index=False, header=False, mode='a+')