来自 Groupby Python 的具有多列的数据透视表

Question

在对（id、gender 和 job）进行 groupby 操作后，我有这个：

id        geneder      job           salary
22         male       engineer         100
                      doctor           120
                       ...
           female     engineer          90
                       doctor          100
                       ...

23         male       engineer         200
                      doctor           150
                       ...
           female     engineer          80
                       doctor          100

从这里我想创建类似的东西：

id       m_engineer    m_doctor ... f_engineer    f_doctor
22          100          120           90           100
23          200          150           80           100

我尝试使用数据透视表，但仍然出现错误。 有什么想法吗？

Answer 1

您需要创建一个可以旋转的新列，例如

(grouped_df
 .reset_index()
 .assign(gender_job = lambda df: df['gender'].str[0] + '_' + df['job'])
 .pivot('id', 'gender_job', 'salary'))