import os
import numpy as np
#取文件夹里图片名字的算法
def file_name(file_dir):
for root, dirs, files in os.walk(file_dir):
# 当前路径下所有子目录
files
#print(np.shape(files),type(files),files) # 当前路径下所有非目录子文件
return files
holiday_dir=file_name('G:\Holiday\jpg1\jpg\')
holiday_dir1=file_name('G:\Holiday\jpg2\jpg\')
print(holiday_dir)
print(holiday_dir[0],type(holiday_dir[0]))
for i in range(len(holiday_dir)):
holiday_dir[i]=holiday_dir[i]+'\n'
for i in range(len(holiday_dir1)):
holiday_dir1[i]=holiday_dir1[i]+'\n'
print(holiday_dir)
f=open('G:\Holiday\holidayName.txt','a')
f.writelines(holiday_dir)
f.close()
f=open('G:\Holiday\holidayName.txt','a')
f.writelines(holiday_dir1)
f.close()
print('文件名提取完成……')import os
import numpy as np
#取文件夹里图片名字的算法
de