Python中如何编写if-else分支下简洁Pythonic的assert断言

你原本写的if/else分支配assert的逻辑是完全正确的，如果要写得更紧凑简洁，可以利用逻辑蕴含的等价转换规则+Python逻辑运算的短路特性，把两个分支的校验规则合并到单个assert语句中，和原写法100%逻辑等价：

assert ("mean" not in d or len(d["mean"]) == N) and ("mean" in d or sigma is not None)

写法说明 #

• 逻辑完全匹配需求，无漏洞：
  两个校验规则是必须同时满足的约束：
  • 如果字典中存在mean键，对应值长度必须等于N：逻辑上「A成立则B必须成立」等价于「A不成立 或 B成立」，对应表达式"mean" not in d or len(d["mean"]) == N
  • 如果字典中不存在mean键，sigma不能为None：同理对应表达式"mean" in d or sigma is not None
• 不会触发KeyError，也不会做多余校验：Python逻辑运算符从左到右求值、遇到确定结果就短路的特性，保证：
  • 当"mean"不在字典中时，第一个表达式的"mean" not in d为True，不会执行后面的len(d["mean"])取值；第二个表达式会走到sigma is not None的校验
  • 当"mean"在字典中时，第二个表达式的"mean" in d为True，不会执行后面的sigma校验；第一个表达式会走到len(d["mean"]) == N的校验
• 如果需要自定义断言失败的报错提示，可以直接在语句后追加提示内容：

assert ("mean" not in d or len(d["mean"]) == N) and ("mean" in d or sigma is not None), "参数非法：要么传入长度为N的mean值，要么保证sigma不为None"

额外注意 #

如果该校验逻辑需要在生产环境生效，不建议使用assert实现——Python以优化模式（加-O参数启动）运行时，会全局移除所有assert语句，导致校验完全失效。这种场景建议显式抛出异常：

if not (("mean" not in d or len(d["mean"]) == N) and ("mean" in d or sigma is not None)):
    raise ValueError("参数配置错误：必须提供长度匹配的mean值，或传入非None的sigma参数")

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