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