Pandas 将级别的每个类别的中间值以外的所有值替换为空白

Question

给定以下枢轴table：

df=pd.DataFrame({'A':['a','a','a','a','a','b','b','b','b'],
                 'B':['x','y','z','x','y','z','x','y','z'],
                 'C':['a','b','a','b','a','b','a','b','a'],
                 'D':[7,5,3,4,1,6,5,3,1]})
table = pd.pivot_table(df, index=['A', 'B','C'],aggfunc='sum')
table

            D
A   B   C   
a   x   a   7
        b   4
    y   a   1
        b   5
    z   a   3
b   x   a   5
    y   b   3
    z   a   1
        b   6

我知道我可以访问每个级别的值:

In [128]:    
table.index.get_level_values('B')

Out[128]:
Index(['x', 'x', 'y', 'y', 'z', 'x', 'y', 'z', 'z'], dtype='object', name='B')

In [129]:
table.index.get_level_values('A')

Out[129]:
Index(['a', 'a', 'a', 'a', 'a', 'b', 'b', 'b', 'b'], dtype='object', name='A')

接下来，我想用空白 ('') 替换每个外部级别中的所有值，中间值或 n/2+1 值除外。

这样：

Index(['x', 'x', 'y', 'y', 'z', 'x', 'y', 'z', 'z'], dtype='object', name='B')

变为：

Index(['x', '', 'y', '', 'z', 'x', 'y', 'z', ''], dtype='object', name='B')

和

Index(['a', 'a', 'a', 'a', 'a', 'b', 'b', 'b', 'b'], dtype='object', name='A')

变为：

Index(['', '', 'a', '', '', '', 'b', '', ''], dtype='object', name='A')

最终，我将尝试将它们用作 Matplotlib 水平条中的二级和三级 y 轴标签，如下图所示（尽管我的一些标签可能会向上移动）：

Answer 1

终于花时间弄明白了...

#First, get the values of the index level.
A=table.index.get_level_values(0)

#Next, convert the values to a data frame.
ndf = pd.DataFrame({'A2':A.values})

#Next, get the count of rows per group.
ndf['A2Count']=ndf.groupby('A2')['A2'].transform(lambda x: x.count())

#Next, get the position based on the logic in the question.
ndf['A2Pos']=ndf['A2Count'].apply(lambda x: x/2 if x%2==0 else (x+1)/2)

#Next, order the rows per group.
ndf['A2GpOrdr']=ndf.groupby('A2').cumcount()+1

#And finally, create the column to use for plotting this level's axis label.
ndf['A2New']=ndf.apply(lambda x: x['A2'] if x['A2GpOrdr']==x['A2Pos'] else "",axis=1)
ndf

    A2  A2Count  A2Pos  A2GpOrdr   A2New
0   a   5        3.0       1    
1   a   5        3.0       2    
2   a   5        3.0       3       a
3   a   5        3.0       4    
4   a   5        3.0       5    
5   b   4        2.0       1    
6   b   4        2.0       2       b
7   b   4        2.0       3    
8   b   4        2.0       4

Pandas 将级别的每个类别的中间值以外的所有值替换为空白

Pandas Replace All But Middle Values per Category of a Level with Blank

python-3.x

pandas

matplotlib

multi-index