本文介绍如何在 pandas 中基于日期时间顺序，为不同标签（如 tag == 2 和 tag != 2）分别计算动态累积和，并生成两个独立的运行总量列（rbq 和 rsq）。
本文介绍如何在 pandas 中基于日期时间顺序，为不同标签（如 tag == 2 和 tag != 2）分别计算动态累积和，并生成两个独立的运行总量列（rbq 和 rsq）。
在实际数据分析中，常需对时间序列数据按业务逻辑进行条件性累积汇总——例如，仅对特定类别（如 Tag == 2）累加，其余类别单独累计，且所有累计必须严格遵循 DateTime 的自然时序。若直接使用 groupby().cumsum() 或未排序就计算，结果将偏离预期（如示例中 RBQ 应随所有非 Tag=2 的记录递增，RSQ 仅累加 Tag=2 的记录，且二者均需按时间先后更新）。
✅ 正确解法的关键在于：先按 DateTime 排序 → 再基于布尔条件构造掩码 → 分别应用 cumsum()。
以下是推荐的简洁、高效实现：
# 确保 DateTime 为 datetime 类型（如有必要）
df['DateTime'] = pd.to_datetime(df['DateTime'])
# 按时间升序排序（关键步骤！）
df_sorted = df.sort_values('DateTime').reset_index(drop=True)
# 定义条件：Tag == 2
cond = df_sorted['Tag'] == 2
# RBQ：仅累加 Tag ≠ 2 的 Qty；Tag == 2 时贡献为 0
df_sorted['RBQ'] = df_sorted['Qty'].mask(cond, 0).cumsum()
# RSQ：仅累加 Tag == 2 的 Qty；其余为 0
df_sorted['RSQ'] = df_sorted['Qty'].where(cond, 0).cumsum()
# 若需保留原始行序，可选：用原始索引重新映射（见下方注意事项）? 输出效果（与目标完全一致）： | DateTime | Tag | Qty | RBQ | RSQ | |-------------------|-----|-----|-----|-----| | 2025-01-01 13:00 | 1 | 270 | 270 | 0 | | 2025-01-03 13:22 | 1 | 32 | 302 | 0 | | 2025-01-10 12:33 | 2 | 44 | 302 | 44 | | 2025-01-22 10:04 | 2 | 120 | 302 | 164 | | 2025-01-29 09:30 | 3 | 182 | 484 | 164 | | 2025-02-02 15:05 | 1 | 216 | 600 | 164 |
⚠️ 重要注意事项：
-
排序不可省略：Pandas 的
cumsum()是按 DataFrame 当前行序执行的。若原始数据未按时间排序（如示例中第6行时间早于第5行），直接计算会导致 RBQ/RSQ 逻辑错乱。务必先sort_values('DateTime')。 -
避免链式赋值：推荐先排序生成新 DataFrame（如
df_sorted），再添加列，防止 SettingWithCopyWarning。 -
类型一致性：
mask()和where()返回 float 类型（因填充值0为浮点），如需整数列，可追加.astype(int)（但需确保无 NaN）。 -
扩展性提示：若需支持多标签（如 Tag ∈ {1,3} 归 RBQ，Tag == 2 归 RSQ，Tag == 4 归 RTQ），可用
np.select()构造多条件数组，再分列cumsum()。
该方法兼具可读性、性能与健壮性，是 Pandas 时间条件累积求和的标准实践。
