csv文件的输出格式 - Python

Question

来自具有以下格式的csv文件：

Date,Data
01-01-01,111
02-02-02,222
03-03-03,333

我使用以下代码计算每月的平均值：

data = pd.read_csv("input.csv")
data['Month'] = pd.DatetimeIndex(data.reset_index()['Date']).month
mean_data = data.groupby('Month').mean()

然后我想在新的csv文件上写出输出的每月值。我目前正在使用以下代码行：

 with open ("test.csv", "w") as f:
     print(mean_data, file=f)

不幸的是，这产生了一个奇怪的输出，具有以下格式：

      Data
Month
1        2
2        3
3        4
4        5

我想获得一个真正的csv输出如下：

Month,Data
1,2
2,3
3,4
4,5

有谁知道如何实现这一目标？

Answer 1

而不是使用：

 with open ("test.csv", "w") as f:
    print(mean_data, file=f)

您可以尝试使用：

import csv

header = ['Month', 'Data']
with open("test.csv", "w") as f:
    writer = csv.writer(f)
    writer.writerow(header)
    writer.writerows(mean_data)

我尝试重新创建您的问题，上面应该提供解决方案。

Answer 2

您可以使用mean_data.to_csv(...)

以下是文档：

Pandas to_csv