熊猫 groupby 和 pivot

Question

我有以下熊猫数据框

   id   category    counts_mean
0   8   a           23
1   8   b           22
2   8   c           23
3   8   d           30
4   9   a           40
5   9   b           22
6   9   c           11
7   9   d           10
....

我想按 id 分组并转置类别列以获得如下内容：

  id  a  b  c  d  
0 8   23 22 23 30
1 9   40 22 11 10

我用 groupby 和 pivot 尝试了不同的东西，但我不确定 groupby 的聚合参数应该是什么......

Answer 1

您只需要使用 pivot 函数并设置参数（索引、列、值）来重新塑造您的 DataFrame，而不是使用 groupby 和 pivot。

#Creat the DataFrame

data = {
'id': [8,8,8,8,9,9,9,9],
'catergory': ['a','b','c','d','a','b','c','d'],
'counts_mean':  [23,22,23,30,40,22,11,10]
}

df = pd.DataFrame(data)

# Using pivot to reshape the DF

df_reshaped = df.pivot(index='id',columns='catergory',values = 'counts_mean')
print(df_reshaped) 

output:
catergory   a   b   c   d
id                       
8          23  22  23  30
9          40  22  11  10