时间戳

首页 > 解决方案 > 从多索引熊猫数据框创建索引

python - 从多索引熊猫数据框创建索引

问题描述

我有一个数据框

   0        1      
   A   B    A    B   
1  4   6    5    6  
2  4   8    15   3
3  8   10   10   2

我想创建一个索引，因此所有列都除以 A 列的总和，即它们的透视级别 0 即 0 或 1

所以

0=16
1=30

所以输出是

   0              1      
   A   B          A       B   
1  4/16   6/16    5/30    6/30  
2  4/16   8/16    15/30   3/30
3  8 /16  10/16   10/30   2/30

到目前为止我有

df_in.divide(df_in.iloc[:, df_in.columns.get_level_values(1)=='A'].sum(axis=0),level=0,axis=1)

但它不工作

标签： pythonpandasmulti-index

解决方案

用于DataFrame.xs选择级别A并sum传递给DataFrame.div：

df = df.div(df.xs('A', axis=1, level=1).sum(), axis=1, level=0)
print (df)
      0                1          
      A      B         A         B
1  0.25  0.375  0.166667  0.200000
2  0.25  0.500  0.500000  0.100000
3  0.50  0.625  0.333333  0.066667

推荐阅读