写入到csv文件的两种方式（pd.DaaFrame 和 csv.writerow）

第一种：

 pd.DataFrame
 to_csv

tmp = pd.DataFrame({"id":[str(i) for i in range(len(test_x))],f"{name}":outputs})
print("save csv ...")
tmp.to_csv(os.path.join(path_prefix, name+'_predict.csv'), index=False)

 

第二种：

csv_writer.writerow

import csv
sumbmit_csv_path = "submit_have_valid_SGD.csv"

with open(sumbmit_csv_path, "w") as f:
    csv_writer = csv.writer(f)
    header = ['id', 'value']
    print(header)
    csv_writer.writerow(header)
    for i in range(240):
        row = ['id_' + str(i), y_test[i][0]]
        csv_writer.writerow(row)
        
        print(row)
print("over")