def get_Write_file_infos(path):
# 文件信息列表
file_lists = []
labels = 0
# 遍历并写入文件信息
for root, dirnames, filenames in os.walk(path):#
# print(‘root’,root,’ dirnames’, dirnames, ‘filenames’, filenames)
for filename in filenames:
# print(filenames)
# file_infos = {}
# dirname = root
# dirname = re.findall(r’val\(.+)’, dirname)
# # 正则表达式来截取文件夹名称
# # 正则表达式还得学习
# filename1 = filename.split(’.jpg’)[0]
# file_infos[“file”] = filename1
# file_infos[“species”] = dirname[0]
# file_infos[“图片”] = ‘’
# 将数据追加字典到列表中
curr_file = root.split(os.sep)[-1] # 表示分离出路径下倒数第一块,os.sep相当于路径中的“/”或“\”
# print(curr_file)
if curr_file == ‘neg’:
labels = 0
elif curr_file == ‘pos’:
labels = 1
else:
labels = 2
file_lists.append([os.path.join(curr_file, filename), labels])
return file_lists
写入csv文件
def write_csv(file_lists):
with open(‘1.csv’, ‘w’, newline=’’) as csvfile:
# writer = csv.DictWriter(csvfile, fieldnames=[‘path’, ‘labels’])
writer = csv.writer(csvfile)#, fieldnames=[‘path’, ‘labels’]
# writer.writeheader()
writer.writerow([‘path’, ‘label’])
for row in file_lists:
writer.writerow(row)