同时将变量存入csv文件

import pandas as pd
import numpy as np

#创建csv文件
df = pd.DataFrame(columns = ['acc', 'nmi', 'ari', 'f1'])
df.to_csv("metrics.csv")

for epoch in range(max_epoch):
    if epoch % 1 ==0:
        '''
        
        '''
        acc, nmi, ari, f1 = eva(y, label)
        metrics = [acc, nmi, ari, f1]   
        #关键: [metrics], mode='a'
        #数据保存在一维列表
        metrics = pd.DataFrame([metrics])
        metrics.to_csv("metrics.csv", mode='a', header=False, index = False)

#几个关键点:

#提前创建csv文件

#数据保存在一维列表

# mode='a'

Guess you like

Origin blog.csdn.net/ziqingnian/article/details/120683184