Python：将分组均值分配给1-D数组

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python：将分组均值分配给1-D数组，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含678字，纯文字阅读大概需要1分钟。

内容图文

假设我有2个数组：

x    = [2, 4, 1, 7, 3, 9, 2, 5, 5, 1]
flag = [0, 1, 0, 2, 1, 1, 2, 0, 0, 2]

标志数组指示x的每个元素属于哪个“组”.如何用x的所有元素的平均值替换x的每个元素(比如标志值k),其对应的标志值也是k？

经过这样的转换后,x看起来像：

x    = [3.25, 5.33, 3.25, 3.33, 5.33, 5.33, 3.33, 3.25, 3.25, 3.33]

(我可以使用循环来实现这一点,但效率很低.)

解决方法:

一种选择是使用熊猫：

import pandas as pd
x    = [2, 4, 1, 7, 3, 9, 2, 5, 5, 1]
flag = [0, 1, 0, 2, 1, 1, 2, 0, 0, 2]
s = pd.Series(x,index=flag)
s.groupby(level=0).transform('mean').tolist()

输出：

[3.25,
 5.333333333333333,
 3.25,
 3.3333333333333335,
 5.333333333333333,
 5.333333333333333,
 3.3333333333333335,
 3.25,
 3.25,
 3.3333333333333335]

内容总结

以上是互联网集市为您收集整理的Python：将分组均值分配给1-D数组全部内容，希望文章能够帮你解决Python：将分组均值分配给1-D数组所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端