Python中如何统一校验反三角函数输入的合法取值范围

你不需要逐行给反三角函数调用加判断，两种通用方案可以一次编写、全局生效： #

你提供的示例代码如下（注：原代码存在运算优先级问题，后续方案中会标注）：

from math import * 

def f(a,b,c,d): 
    alpha = arccos(a/d)
    beta = pi - arcsin(a+b/d) 
    theta= arccos(a/c)
    eta= arccos (a+c/b+d) 
    return (alpha, beta, theta, eta)

方案1：封装安全版本的反三角函数（推荐，逻辑可控无侵入风险） #

只需要写一次范围校验逻辑，封装出带校验的安全版反三角函数，后续调用直接用安全版即可，不需要逐行加if判断：

import math

def _validate_trig_input(x):
    # 两种校验逻辑二选一：
    # 1. 适配浮点计算误差：把越界参数钳位到[-1, 1]合法区间（适合绝大多数数值计算场景）
    return max(-1.0, min(1.0, x))
    # 2. 严格校验：参数越界直接抛出明确异常（适合对输入合法性要求高的场景）
    # if not (-1.0 <= x <= 1.0):
    #     raise ValueError(f"反三角函数参数{x}超出合法范围[-1,1]")

# 封装安全版函数
def safe_asin(x):
    return math.asin(_validate_trig_input(x))

def safe_acos(x):
    return math.acos(_validate_trig_input(x))

# 原有业务代码只需要把arcsin/arccos替换为safe开头的版本即可
# 注意原代码的运算优先级问题：a+b/d 实际是先算b/d再加a，如果要先算a+b再除以d需要加括号写成(a+b)/d
def f(a,b,c,d): 
    alpha = safe_acos(a/d)
    beta = math.pi - safe_asin(a + b/d)
    theta = safe_acos(a/c)
    eta = safe_acos(a + c/b + d)
    return (alpha, beta, theta, eta)

方案2：猴子补丁替换原生函数（适合已有大量历史代码的场景） #

如果不想修改任何已有业务代码的调用逻辑，可以直接在程序入口处替换math模块的原生反三角函数，后续所有调用原生asin/acos的位置都会自动走校验逻辑：

import math

# 先保存原生函数的引用
_origin_asin = math.asin
_origin_acos = math.acos

def _validate_trig_input(x):
    # 校验逻辑和方案1一致，按需选择钳位或者抛错
    return max(-1.0, min(1.0, x))

# 定义打补丁后的函数
def patched_asin(x):
    return _origin_asin(_validate_trig_input(x))

def patched_acos(x):
    return _origin_acos(_validate_trig_input(x))

# 替换原生函数
math.asin = patched_asin
math.acos = patched_acos

# 注意：如果用from math import * 导入，需要在打补丁之后再执行导入，否则补丁不会生效

注：提问中的示例函数为随机编写，无实际业务含义。

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