枢轴熊猫数据框具有多索引列

Question

我正在努力解决如何将数据框旋转为具有多索引列。

我有一个这样的数据框：

data = pd.DataFrame({"name":["a", "a", "b", "b", "b", "b",  "c", "c"], 
                     "month":[1, 1, 2, 3, 1, 1, 1, 3], 
                     "buy_sell":["sell", "buy", "sell", "buy", "sell", "buy", "sell", "buy"],
                     "value":[10, 20, 30, 40, 20, 80, 50, 60]})
data

我想将其转换为宽格式。索引是name，对于列我想用month和buy_sell的组合创建多索引

或者，如果不是多索引列，我想旋转数据框，以便列具有足够的空间，例如 sell_1、buy_1、sell_2、buy_2 等...

任何帮助将不胜感激。谢谢！

Answer 1

您可以使用set_index() 和unstack()：

(data.set_index(['name','month','buy_sell'])['value']
      .unstack(['month','buy_sell']))

输出：

month        1           2     3
buy_sell  sell   buy  sell   buy
name                            
a         10.0  20.0   NaN   NaN
b         20.0  80.0  30.0  40.0
c         50.0   NaN   NaN  60.0

如果您坚持让这些列包含所有NaN，您可以unstack 一次一级：

(data.set_index(['name','month','buy_sell'])['value']
      .unstack('month').unstack('buy_sell')
     )

输出：

month        1         2           3     
buy_sell   buy  sell buy  sell   buy sell
name                                     
a         20.0  10.0 NaN   NaN   NaN  NaN
b         80.0  20.0 NaN  30.0  40.0  NaN
c          NaN  50.0 NaN   NaN  60.0  NaN