You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Pandas按客户分组移位列:首日值设为0的前一日计数列需求

为透视后的Pandas DataFrame添加previous_day_count列

嘿,这个需求我太熟了!要给每个客户新增previous_day_count列,而且客户首日的前一日计数值设为0,用Pandas的分组+移位操作就能轻松搞定,下面给你一步步拆解:

步骤1:确保数据排序正确

移位操作的逻辑完全依赖数据按客户和日期的顺序排列,所以先做排序保证后续逻辑准确:

# 按客户名称和当前日期排序,为移位操作打好基础
df = df.sort_values(by=['customer_name', 'current_date'])

步骤2:分组移位并填充首日值

用groupby按客户分组,然后对current_day_count执行移位操作,最后把首日产生的NaN替换成0:

# 分组后移位,自动处理每个客户的首日值为0
df['previous_day_count'] = df.groupby('customer_name')['current_day_count'].shift(1).fillna(0)

示例效果

假设你的原始透视后数据是这样的:

customer_namecurrent_datecurrent_day_count
Alice2024-01-015
Alice2024-01-027
Bob2024-01-013
Bob2024-01-034

执行代码后会得到:

customer_namecurrent_datecurrent_day_countprevious_day_count
Alice2024-01-0150
Alice2024-01-0275
Bob2024-01-0130
Bob2024-01-0343

简单解释下:shift(1)会把每个客户组内的current_day_count向下挪一行,所以每组的第一行就会变成NaN,用fillna(0)刚好把这些NaN替换成需求里要求的0,完美踩中需求点!

内容的提问来源于stack exchange,提问作者bala chandar

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.19 07:48:05