pandas中如何对满足特定条件的行执行条件列计算

以下是两种常用的实现方式，均不会触发副本警告：

1. 计算结果存入原表新列 #

如果需要将计算结果保留在原DataFrame中，不符合条件的行对应字段自动填充NaN：

# 仅snapshot为'2021-02-02'的行执行比值计算，结果存入ratio列
df.loc[df['snapshot'] == '2021-02-02', 'ratio'] = df['numerator'] / df['denominator']

2. 临时获取目标行的计算结果 #

不需要修改原表，只返回符合条件行的比值序列：

target_ratio = df.loc[df['snapshot'] == '2021-02-02', 'numerator'] / df.loc[df['snapshot'] == '2021-02-02', 'denominator']

异常场景处理 #

如果denominator列存在0值，会导致计算结果为inf，可叠加分母非0的过滤条件规避：

df.loc[
    (df['snapshot'] == '2021-02-02') & (df['denominator'] != 0),
    'ratio'
] = df['numerator'] / df['denominator']

内容的提问来源于stack exchange，提问作者Daven1