python - 如何将屏幕输出的结果保存到 python 上 2 列中的 csv 文件？

Question

import pandas as pd
import numpy as np
import csv

with open('predictions.csv', 'r') as f:
    csvreader = csv.reader(f)
    next(csvreader)
    for r in csvreader:
        df = r[:-1]
        a = float(str(df[:][0]))
        b = float(str(df[:][1]))
        if a > b :
            print(1, 0)
        else:
            print(0, 1)

0 1
0 1
0 1
0 1
0 1
0 1

我从 predictions.csv 文件中打印了我的结果，并尝试将显示为 0 和 1 的结果保存到 2 列中的 CSV 文件中。那么如何将我的结果保存到 2 列的 CSV 文件中？

标签： pythonpython-3.xcsv

Answer 1

您不必保存打印的输出 - 当您在循环中打印数据时，可以直接将数据保存到 CSV，如下所示：

with open('new_file.csv', mode='a', newline='') as file:
    csv_writer = csv.writer(file)
    if a > b :
        print(1, 0)
        csv_writer.writerow(['1', '0'])
    else:
        print(0, 1)
        csv_writer.writerow(['0', '1'])