如何在Python中重塑此DataFrame？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了如何在Python中重塑此DataFrame？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1059字，纯文字阅读大概需要2分钟。

内容图文

我想重塑Python中的DataFrame df_sale,在价格列中计算总和并添加一个新的总计.以下是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和231使用sum().我不知道如何重塑数据,添加新的列标题并获得总数而又不会降低计算效率.谢谢.

解决方法:

这可能不是最干净的方法,但是可以得到您想要的结果：

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

内容总结

以上是互联网集市为您收集整理的如何在Python中重塑此DataFrame？全部内容，希望文章能够帮你解决如何在Python中重塑此DataFrame？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端