本文介绍了如何使用 Pandas 计算 DataFrame 中，按照 ID 分组后，每一行数据对应的前序行的中位数。通过 groupby.transform、shift 和 expanding.median 函数的组合，可以高效地实现这一目标，无需手动循环，代码简洁易懂。
在数据分析中，经常需要计算基于时间序列或分组数据的滚动统计量。本文将演示如何使用 Pandas 计算 DataFrame 中，按照特定 ID 分组后，每一行数据对应的前序行的中位数。
问题描述
假设我们有一个 DataFrame，包含 Index、ID 和 Amount 三列，如下所示：
Index ID Amount0 1 A 101 2 A 152 3 A 173 4 A 124 5 A 105 6 B 206 7 B 15
我们的目标是添加一个新的列 MedianOfPastElements，其中每一行的值是相同 ID 的所有前序行的 Amount 列的中位数。
解决方案
Pandas 提供了强大的 groupby 和 transform 功能，可以高效地解决这个问题。以下是实现的代码：
import pandas as pd# 示例数据data = {'Index': [1, 2, 3, 4, 5, 6, 7], 'ID': ['A', 'A', 'A', 'A', 'A', 'B', 'B'], 'Amount': [10, 15, 17, 12, 10, 20, 15]}df = pd.DataFrame(data)# 计算前序行的中位数df['MedianOfPastElements'] = (df.groupby('ID')['Amount'] .transform(lambda s: s.shift().expanding().median()))print(df)
代码解释
算家云
高效、便捷的人工智能算力服务平台
37 查看详情 df.groupby(‘ID’)[‘Amount’]: 首先，我们按照 ID 列对 DataFrame 进行分组，并选择 Amount 列。.transform(lambda s: …): transform 函数会将一个函数应用到每个分组的 Series 上，并将结果合并回原始 DataFrame。s.shift(): shift() 函数将 Series 中的元素向下移动一位。这样，每一行就对应了它前面所有行的值。例如，对于 ‘A’ 组的第三行，shift() 后的值为前两行的值。.expanding(): expanding() 函数创建一个 expanding object，允许我们计算累积统计量。.median(): median() 函数计算每个 expanding window 的中位数。
输出结果
运行上述代码，将得到以下结果：
Index ID Amount MedianOfPastElements0 1 A 10 NaN1 2 A 15 10.02 3 A 17 12.53 4 A 12 15.04 5 A 10 13.55 6 B 20 NaN6 7 B 15 20.0
可以看到，MedianOfPastElements 列包含了我们想要的结果。第一行由于没有前序行，所以中位数为 NaN。
注意事项
shift() 函数会导致第一行的值为 NaN，因为没有前序行。如果需要，可以使用 fillna() 函数填充这些缺失值。该方法适用于大型 DataFrame，因为它使用了 Pandas 的矢量化操作，效率很高。可以根据需要修改代码，计算其他统计量，例如平均值、标准差等。只需要将 .median() 替换为 .mean()、.std() 等相应的函数即可。
总结
本文演示了如何使用 Pandas 的 groupby.transform、shift 和 expanding.median 函数，高效地计算 DataFrame 中，按照 ID 分组后，每一行数据对应的前序行的中位数。这种方法简洁、高效，适用于处理大型数据集。掌握这些技巧可以帮助你更有效地进行数据分析和处理。
以上就是Pandas 计算同 ID 下前序行的中位数的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/597910.html
微信扫一扫 
支付宝扫一扫 
