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

Python Pandas:按条件传递用户数值(替换来源B数据)的实现问询

Pandas按用户分组实现数值传递操作

要实现你需要的逻辑,可按以下步骤操作:

  • 先将Value列转换为数值类型(示例中该列为字符串,避免后续运算出错)
  • 按USER分组,生成每组内上一行的Value
  • 通过条件判断完成替换:仅当Source为B且当前行不是用户组内首行时,用上一行Value替换,其余情况保留原值

完整代码示例

import pandas as pd
import numpy as np

# 初始化数据
df = pd.DataFrame({
    "USER": ["USER1", "USER1", "USER1", "USER1", "USER1", "USER2", "USER2", "USER2", "USER2", "USER2"],
    "Date": ["01/01/2023", "01/02/2023", "02/02/2023", "20/02/2023", "01/07/2023", "01/01/2023", "01/02/2023", "02/02/2023", "20/02/2023", "01/07/2023"],
    "Value": ["1000", "1200", "1300", "1000", "1400", "1000", "1200", "1250", "1300", "1400"],
    "Source": ["A", "A", "A", "B", "A", "A", "A", "B", "A", "A"]
})

# 转换Value为数值类型
df['Value'] = df['Value'].astype(int)

# 按用户分组,获取上一行的Value
df['prev_value'] = df.groupby('USER')['Value'].shift(1)

# 条件替换Value
df['Value'] = np.where(
    (df['Source'] == 'B') & df['prev_value'].notna(),
    df['prev_value'],
    df['Value']
)

# 删除临时列
df = df.drop('prev_value', axis=1)

# 查看结果
print(df)

运行结果

USERDateValueSource
USER101/01/20231000A
USER101/02/20231200A
USER102/02/20231300A
USER120/02/20231300B
USER101/07/20231400A
USER201/01/20231000A
USER201/02/20231200A
USER202/02/20231200B
USER220/02/20231300A
USER201/07/20231400A

完全匹配你给出的目标数据要求。

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

相关产品推荐
方舟 Agent Plan

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

最近更新时间:2026.07.09 16:25:17