python - 如何在 Pandas 数据透视表中获取交替值列？

Question

我想从 Pandas DataFrame 生成一个报告，该报告使用交替值列而不是交替主列。例如，给定以下代码:

import pandas as pd
data = [('a1', 'b1', 1, 2),
        ('a1', 'b2', 3, 4),
        ('a1', 'b3', 5, 6),
        ('a2', 'b1', 7, 8),
        ('a2', 'b2', 9, 10),
        ('a2', 'b3', 11, 12)]

df = pd.DataFrame(data, columns=['A', 'B', 'X', 'Y'])
print(pd.pivot_table(df, index='A', columns=['B']))

我得到这个输出:

    X         Y        
B  b1 b2  b3 b1  b2  b3
A                      
a1  1  3   5  2   4   6
a2  7  9  11  8  10  12

但是，我更喜欢这样的东西:

B  b1    b2      b3 
    X  Y  X   Y   X  Y        
A                      
a1  1  2  3   4   5   6
a2  7  8  9  10  11  12

有什么合理的方法可以实现这一目标吗？

Answer 1

要做的第一件事是交换 MultiIndex 数据帧上的级别。然后你必须对其进行排序。

In [1]: pd.pivot_table(df, index='A', columns=['B']).swaplevel(axis=1).sort_index(axis=1)
Out[1]:
B  b1    b2      b3
     X  Y  X   Y   X   Y
A
a1  1  2  3   4   5   6
a2  7  8  9  10  11  12