如何将Python的if-elif分支代码改写为合法的单行实现形式

可行的单行改写方案 #

你给出的示例写法无法运行，是因为Python的三元条件表达式仅支持包含表达式，不能直接将赋值语句作为分支内容。以下是几种合法的改写方式，可根据场景选择：

• 简洁方案（仅适用于x取值仅为0/1的场景）：通过条件表达式匹配变量名，从局部命名空间取变量操作

locals()["a" if x == 0 else "b"] += j["sum"]

• 通用方案（支持x存在其他取值的场景，无额外判断开销）：分号分隔两个带条件的增量赋值

a += j["sum"] if x == 0 else 0; b += j["sum"] if x == 1 else 0

说明：该写法仅当x匹配对应值时才会执行增量操作，其他取值下两个变量都不会变更，和原代码逻辑完全一致。

• 分支对齐方案（完全复刻原代码的elif判断逻辑，x匹配到0后不会再判断x==1）：嵌套条件表达式结合exec实现

exec('a += j["sum"]' if x == 0 else 'b += j["sum"]' if x == 1 else '')

注意：exec会执行动态生成的代码字符串，存在潜在安全风险，仅推荐输入完全可控的场景使用。

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