python - 如何在 Python 中 reshape 此 DataFrame？

Question

我在 Python 中有一个 DataFrame df_sale，我想 reshape 它，计算 price 列的总和并添加一个新的列 total .以下是 df_sale:

b_no  a_id  price  c_id
120   24     50     2
120   56     100    2
120   90     25     2
120   45     20     2
231   89     55     3
231   45     20     3
231   10     250    3

整形后异常输出:

b_no  a_id_1  a_id_2  a_id_3  a_id_4  total  c_id
120   24      56      90      45      195    2
231   89      45      10      0       325    3

到目前为止，我已经尝试过在 df_sale['price'] 上分别对 120 和 使用 sum() 231。我不明白我应该如何 reshape 数据、添加新的列标题并在计算效率不高的情况下获得总数。谢谢。

Answer 1

这可能不是最干净的方法(根本)，但它会得到你想要的结果:

reshaped_df = (df.groupby('b_no')[['price', 'c_id']]
               .first()
               .join(df.groupby('b_no')['a_id']
                     .apply(list)
                     .apply(pd.Series)
                     .add_prefix('a_id_'))
               .drop('price',1)
               .join(df.groupby('b_no')['price'].sum().to_frame('total'))
               .fillna(0))


>>> reshaped_df
      c_id  a_id_0  a_id_1  a_id_2  a_id_3  total
b_no                                             
120      2    24.0    56.0    90.0    45.0    195
231      3    89.0    45.0    10.0     0.0    325