python - Pandas 由长到宽的 reshape ，通过两个变量

Question

我有长格式的数据，并且正在尝试将其 reshape 为宽格式，但似乎没有一种简单的方法可以使用melt/stack/unstack来实现此目的:

Salesman  Height   product      price
  Knut      6        bat          5
  Knut      6        ball         1
  Knut      6        wand         3
  Steve     5        pen          2

变成:

Salesman  Height    product_1  price_1  product_2 price_2 product_3 price_3  
  Knut      6        bat          5       ball      1        wand      3
  Steve     5        pen          2        NA       NA        NA       NA

我认为 Stata 可以使用 reshape 命令执行类似的操作。

Answer 1

这是另一个更充实的解决方案，取自 Chris Albon's site 。

创建“长”数据框

raw_data = {'patient': [1, 1, 1, 2, 2],
                'obs': [1, 2, 3, 1, 2],
          'treatment': [0, 1, 0, 1, 0],
              'score': [6252, 24243, 2345, 2342, 23525]}

df = pd.DataFrame(raw_data, columns = ['patient', 'obs', 'treatment', 'score'])

制作“宽”数据

df.pivot(index='patient', columns='obs', values='score')