python - 如何在特定行上旋转 Pandas 数据框

Question

我有以下数据框:

df = pd.DataFrame({'recipe': ['a', 'a', 'a', 'a', 'a','a', 'b', 'b', 'b', 'b', 'b','b'],
                   'product': ['A', 'B', 'C', 'D', 'E', 'F', 'A', 'B', 'G', 'I', 'K', 'F']})

我想将输出的数据框旋转为每个配方一行，每个产品多列，例如:

recipe  p1  p2  p3  p4  p5  p6
a       A   B   C   D   E   F
b       A   B   G   I   K   F

另外，有多种产品，但我希望根据另一列给出产品相关程度排名的前 6 名

Answer 1

这实际上是一个groupby和unnesting问题。

u = df.groupby('recipe')['product'].apply(list)
pd.DataFrame(u.tolist(), index=u.index).rename(lambda x: 'p{}'.format(x+1), axis=1)

       p1 p2 p3 p4 p5 p6
recipe                  
a       A  B  C  D  E  F
b       A  B  G  I  K  F