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_name | current_date | current_day_count |
|---|---|---|
| Alice | 2024-01-01 | 5 |
| Alice | 2024-01-02 | 7 |
| Bob | 2024-01-01 | 3 |
| Bob | 2024-01-03 | 4 |
执行代码后会得到:
| customer_name | current_date | current_day_count | previous_day_count |
|---|---|---|---|
| Alice | 2024-01-01 | 5 | 0 |
| Alice | 2024-01-02 | 7 | 5 |
| Bob | 2024-01-01 | 3 | 0 |
| Bob | 2024-01-03 | 4 | 3 |
简单解释下:shift(1)会把每个客户组内的current_day_count向下挪一行,所以每组的第一行就会变成NaN,用fillna(0)刚好把这些NaN替换成需求里要求的0,完美踩中需求点!
内容的提问来源于stack exchange,提问作者bala chandar
相关产品推荐
相关产品推荐

