格式 Pandas 枢轴 Table

Question

我在格式化由 Pandas 创建的枢轴 table 时遇到了问题。因此，我通过使用 pandas.pivot_table 将 A 作为列，将 B 作为索引，在我的源数据的 2 列 (A,B) 之间创建了一个矩阵 table。

>> df = PD.read_excel("data.xls")
>> table = PD.pivot_table(df,index=["B"],
    values='Count',columns=["A"],aggfunc=[NUM.sum],
    fill_value=0,margins=True,dropna= True)
>> table

它returns为：

      sum
  A     1  2  3  All
  B 
  1     23 52  0  75
  2     16 35 12  65
  3     56  0  0  56
All     95 87 12 196

而且我希望有这样的格式：

                A      All_B
            1   2   3   
    1      23  52   0     75
B   2      16  35  12     65
    3      56   0   0     56
All_A      95  87  12    196

我应该怎么做？非常感谢。

Answer 1

pd.pivot_table 返回的 table 非常方便处理（它是单级 index/column）并且通常不需要任何进一步的格式操作。但是如果你非要改成你在post中提到的格式，那么你就需要用pd.MultiIndex构造一个多级的index/column。这是有关如何操作的示例。

操作前，

import pandas as pd
import numpy as np

np.random.seed(0)
a = np.random.randint(1, 4, 100)
b = np.random.randint(1, 4, 100)
df = pd.DataFrame(dict(A=a,B=b,Val=np.random.randint(1,100,100)))
table = pd.pivot_table(df, index='A', columns='B', values='Val', aggfunc=sum, fill_value=0, margins=True)
print(table)


B       1     2     3   All
A                          
1     454   649   770  1873
2     628   576   467  1671
3     376   247   481  1104
All  1458  1472  1718  4648

之后：

multi_level_column = pd.MultiIndex.from_arrays([['A', 'A', 'A', 'All_B'], [1,2,3,'']])
multi_level_index = pd.MultiIndex.from_arrays([['B', 'B', 'B', 'All_A'], [1,2,3,'']])
table.index = multi_level_index
table.columns = multi_level_column
print(table)

            A             All_B
            1     2     3      
B     1   454   649   770  1873
      2   628   576   467  1671
      3   376   247   481  1104
All_A    1458  1472  1718  4648

格式 Pandas 枢轴 Table

Format Pandas Pivot Table

python

pandas

dataframe

pivot-table