Python, Groupbyの使い方について

Question

Pythonのデータ処理に関する質問です。

以下のようなデータセットをGroupbyで列を追加したいのですが、うまくいきません。
user_idとstatusのkeyが二つあるためうまくいかないのか、そもそもg.aggやg.sizeの範疇ではないのかわからなくなりました。

【元データ】
user_id, status, category, category_count
001, A, Cat1, 1
001, A, Cat2, 2
001, B, Cat3, 3
002, A, Cat1, 4
002, A, Cat2, 5

【変更したい形式】
user_id, status, Cat1_category_count, Cat2_category_count, Cat3_category_count
001, A, 1, 2, 0
001, B, 0, 0, 3
002, A, 4, 5, 0

どなたかよろしくお願いします。

Answer 1

melianさん

的確で情報が詰まったご回答をいただきありがとうございました。
無事ワークしました。

Answer 2

import pandas as pd
import io

csv_data = '''
user_id,status,category,category_count
001,A,Cat1,1
001,A,Cat2,2
001,B,Cat3,3
002,A,Cat1,4
002,A,Cat2,5
'''
df = pd.read_csv(io.StringIO(csv_data))

#
dfx = df.pivot_table(index=['user_id', 'status'], columns='category', values='category_count')\
        .fillna(0, downcast='infer')\
        .rename_axis(columns=None)\
        .add_suffix('_category_count')\
        .reset_index()
dfx['user_id'] = dfx['user_id'].map('{:03d}'.format)

print(dfx)

#
#   user_id status  Cat1_category_count  Cat2_category_count  Cat3_category_count
# 0     001      A                    1                    2                    0
# 1     001      B                    0                    0                    3
# 2     002      A                    4                    5                    0

Comments

[Onimai]

melianさん

的確で情報が詰まったご回答をいただきありがとうございました。
無事ワークしました。