Question

import pandas, io

data = io.StringIO('''Fruit,Color,Count,Price
Apple,Red,3,1.29
Apple,Green,9,0.99
Pear,Red,25,2.59
Pear,Green,26,2.79
Lime,Green,99,0.39
''')
df_unindexed = pandas.read_csv(data)
df = df_unindexed.set_index(['Fruit', 'Color'])
grandtotal=df.sum().to_frame().T
subtotal=df.sum(axis=0,level=[0])

df
Out[830]: 
             Count  Price
Fruit Color              
Apple Red        3   1.29
      Green      9   0.99
Pear  Red       25   2.59
      Green     26   2.79
Lime  Green     99   0.39

grandtotal
Out[831]: 
   Count  Price
0  162.0   8.05

subtotal
Out[832]: 
       Count  Price
Fruit              
Apple     12   2.28
Lime      99   0.39
Pear      51   5.38

1）如何将总计作为顶行附加（保留原始数据帧格式）

1a）另外，我想将行索引移到顶行，以便我可以将这些单元格用于（＆＃39;所有水果＆＃39;，＃39; Total＆＃39;）

2）我如何在每个水果上添加小计？

2a）我有两个索引的小计行

期望的输出：

Fruit   Color   Count   Price
All Fruits  Total   162 8.05
Apple   Subtotal    12  2.28
Apple   Red 3   1.29
Apple   Green   9   0.99
Peer    Subtotal    51  5.38
Pear    Red 25  2.59
Pear    Green   26  2.79
Lime    Subtotal    99  0.39
Lime    Green   99  0.39

Answer 1

使用pd.concat，assign，sum和set_index：

df_out = pd.concat([df,
                    df.sum(level=0).assign(Color='SubTotal')\
                      .set_index('Color', append=True),
                    df.sum().to_frame().T\
                      .assign(Fruit='All Fruit', 
                              Color='Total')\
                      .set_index(['Fruit', 'Color'])])\
           .sort_index()

输出：

                   Count  Price
Fruit     Color                 
All Fruit Total     162.0   8.05
Apple     Green       9.0   0.99
          Red         3.0   1.29
          SubTotal   12.0   2.28
Lime      Green      99.0   0.39
          SubTotal   99.0   0.39
Pear      Green      26.0   2.79
          Red        25.0   2.59
          SubTotal   51.0   5.38

将grandtotal和subtotal行附加到两级行和列索引数据帧

1 个答案: