熊猫：如何从数据框中现有列中的值创建新列？

Question

我有一个像这样的熊猫数据框：

name   category  count 
Tom      A         5
Tom      B         3
Tom      C         2
Eva      A        15
Eva      B        13
Eva      C        12

我想将其转换为此：

Name    Category_A     Category_B      Category_C
Tom         5             3                2
Eva        15            13               12

我该怎么做？

Answer 1

这应该有效：

import pandas as pd

result=pd.pivot_table(df, index='name', columns='category', values='count')

Answer 2

尝试一下：

df = df.pivot(columns=['category'], index='name')
df.columns = df.columns.get_level_values(1)
df.columns = ['category_'+i for i in df.columns]
df.reset_index(inplace=True)
print(df)

  name  category_A  category_B  category_C
0  Eva          15          13          12
1  Tom           5           3           2