在 Python 中将数据框保存为 CSV

Question

我试图将此代码的结果保存为 CSV 文件：

import pandas as pd

df = pd.DataFrame({'ID': ['a01', 'a01', 'a01', 'a01', 'a01', 'a01', 'a01', 'a01', 'a01', 'b02', 'b02','b02', 'b02', 'b02', 'b02', 'b02'],

                   'Row': [1, 1, 1, 2, 2, 2, 3, 3, 3, 1, 1, 2, 2, 3, 3, 3],

                   'Col': [1, 2, 3, 1, 2, 3, 1, 2, 3, 1, 3, 1, 3, 1, 2, 3],

                   'Result': ['p', 'f', 'p', 'p', 'p', 'f', 'p', 'p', 'p', 'p', 'p', 'p', 'f', 'p', 'p', 'p']})

dfs = {}

for n, g in df.groupby('ID'):

    dfs[n] = g.pivot('Row', 'Col', 'Result').fillna('')

    print(f'ID: {n}')

    print(dfs[n])

    print('\n')

    print(dfs[n].stack().value_counts().to_dict())

    print('\n')

我找到了几种方法并尝试将输出（字典形式）保存到 CSV 文件中，但没有成功。有什么想法吗？

PS 这是我找到的方法之一，但我不知道如何根据我的输出来命名列？

with open("Output.csv", "w", newline="") as csv_file:

  cols = ["???????????"] 

  writer = csv.DictWriter(csv_file, fieldnames=cols)

  writer.writeheader()

  writer.writerows(data)

Answer 1

df.to_csv('Output.csv', index = False)

Answer 2

使用pandas data frame abject提供的方法df.to_csv()