Pandas三元运算报Series真值不明确错误该如何优化写法？

报错原因 #

原生Python三元运算仅支持对单个标量做逻辑判断，你的代码中df.MTO_STRAT == ""返回的是布尔类型的Series对象（对应整列每一行的判断结果），Python无法直接对序列类型做整体真值判定（不知道该取任意行为真还是全部行为真作为判断依据），因此触发该报错，和else分支隐藏的or运算无关。

推荐实现写法 #

写法1：numpy.where嵌套（全版本兼容） #

该写法对pandas版本无要求，适配所有环境：

import numpy as np
df['mto_mts'] = np.where(
    df['MTO_STRAT'] == "", 
    "MTO",
    np.where(df['MTO_STRAT'] == "Z5", "MTS", "SS")
)

写法2：case_when（pandas 1.3.0+ 版本适用，可读性最优） #

pandas 1.3.0之后新增的case_when方法专门处理多分支赋值场景，逻辑更清晰易维护：

df['mto_mts'] = df['MTO_STRAT'].case_when(
    caselist=[
        (df['MTO_STRAT'] == "", "MTO"),
        (df['MTO_STRAT'] == "Z5", "MTS")
    ],
    default="SS"
)

写法3：映射字典填充（分支较多时优先用） #

如果后续判断规则会持续扩展，提前定义映射关系的写法扩展性更强：

rule_map = {"": "MTO", "Z5": "MTS"}
df['mto_mts'] = df['MTO_STRAT'].map(rule_map).fillna("SS")

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