这里实现一个demo,将结果写入csv中
import pandas as pd
import os
# 定义一个空列表,用于存放结果
result_list = []
# 创建文件夹,存放输出结果
if not os.path.exists("result_test_csv"):
os.mkdir("result_test_csv")
for i in range(10):
for j in range(10):
result_list.append({"id":i,"lebel":j})
result_df = pd.DataFrame(result_list)
result_df.to_csv("result_test_csv\submission.csv", index=False) # index=None去除索引(等同于index=False),index=True保留索引
根据自己的需要,进行一些修改和完善。