将类别转换为数据框python中的列

Question

我有一个包含两列的数据框。一列包含不同的类别，另一列包含值。

import pandas as pd

data={"category":["Topic1","Topic2","Topic3","Topic2","Topic1","Topic3"], "value":["hello","hey","hi","name","valuess","python"]}

df=pd.DataFrame(data=data)

我想要不同的类别到列中，如下所示。

电流输入：

category    value
  Topic1    hello
  Topic2      hey
  Topic3       hi
  Topic2     name
  Topic1  valuess
  Topic3   python

期望输出：

Topic1  Topic2 Topic3
hello    hey    hi
valuess name    python

我尝试使用转置数据帧但没有得到预期的结果。

Answer 1

您可以使用pandas.concat一起axis=1。这也适用于不匹配的长度。

grouper = df.groupby('category')
df = pd.concat([pd.Series(v['value'].tolist(), name=k) for k, v in grouper], axis=1)

print(df)

    Topic1 Topic2  Topic3
0    hello    hey      hi
1  valuess   name  python