def writeCsv(File,species):
    row = [File,species]
    out = open("data/test.csv", "a", newline="")
    csv_writer = csv.writer(out, dialect="excel")
    csv_writer.writerow(row)
#若想在csv文件第一行加标题 writeCsv("file","species")

#使用的时候直接调用就行,例如我写了一个代码中需要调用它,调用代码是
writeCsv(file, classes[pred.data.item()])

打开CSV文件,就是这样的
在这里插入图片描述

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐