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(...)

这里是文档：

熊猫to_csv