在 Pandas 中使用 Pivot_Table

Question

我在 Python 3.7 中使用 Pandas。 我有一個包含 8 列的 csv 文件，我需要旋轉第 5、6、7 和 8 列。我像這樣旋轉它們：

pivot = pd.pivot_table (csv_file, 
                        values = ["column 5", "column 6" , "column 7","column 8"],
                        index = "ID",
                        aggfunc = np.mean,
                        dropna = False)

然后我將它保存到一個csv。 問題是保存的 csv 文件只有第 5、6、7 和 8 列。我希望它也保留前 4 列。 我試圖找到解決方法，例如從 csv_file 中刪除最后 4 列，然后在旋轉后重新添加這些列。 我嘗試連接、加入和合並，但在旋轉后重新創建包含所有 8 列的 csv 文件沒有成功。 如果有人對如何執行此任務有任何見解或知識，我將不勝感激

Answer 1

我沒有你的 csv_file 作為輸入，但我試圖做這樣的事情。 看看這是否是您要找的。

c1 = [i for i in range(10)]
c2 = [i for i in range(10)]
c3 = [str(i) for i in range (10)]
c4 = [str(i) for i in range (10,0,-1)]
c5 = ['test']*10

import pandas as pd
import numpy as np

df = pd.DataFrame({'column1':c1,'column2':c2,
                   'column3':c3,'column4':c4,'column5':c5,})

df_pivot=pd.pivot_table(df,
                  values=['column1', 'column2'], 
                  index=["column3","column4","column5"], 
                  aggfunc={'column1':[np.mean,max,min],'column2':np.sum})

#df_pivot.columns = df_pivot.columns.get_level_values(1)

df_pivot.columns = ['_'.join(col).strip() for col in df_pivot.columns.values]

print(df)
print(df_pivot)

輸出如下：

初始數據幀：

   column1  column2 column3 column4 column5
0        0        0       0      10    test
1        1        1       1       9    test
2        2        2       2       8    test
3        3        3       3       7    test
4        4        4       4       6    test
5        5        5       5       5    test
6        6        6       6       4    test
7        7        7       7       3    test
8        8        8       8       2    test
9        9        9       9       1    test

樞軸數據框

                         column1_max  column1_mean  column1_min  column2_sum
column3 column4 column5                                                     
0       10      test               0             0            0            0
1       9       test               1             1            1            1
2       8       test               2             2            2            2
3       7       test               3             3            3            3
4       6       test               4             4            4            4
5       5       test               5             5            5            5
6       4       test               6             6            6            6
7       3       test               7             7            7            7
8       2       test               8             8            8            8
9       1       test               9             9            9            9

我有幾分鍾的時間來處理你的問題陳述。 您需要調出要寫入文件的列。 這就是我所做的。

寫入 CSV 文件

df_pivot.to_csv('out.csv', index_label=df_pivot.columns.name)

其輸出如下。 我希望這就是你正在尋找的。

column3,column4,column5,column1_max,column1_mean,column1_min,column2_sum
0,10,test,0,0,0,0
1,9,test,1,1,1,1
2,8,test,2,2,2,2
3,7,test,3,3,3,3
4,6,test,4,4,4,4
5,5,test,5,5,5,5
6,4,test,6,6,6,6
7,3,test,7,7,7,7
8,2,test,8,8,8,8
9,1,test,9,9,9,9

寫入 Excel 文件

import openpyxl

with pd.ExcelWriter('myfile.xlsx',engine="openpyxl") as writer:
    df_pivot.to_excel(writer, sheet_name='Sheet1',columns=df_pivot.columns)

我得到的答案如下：