mbcp/docs/en/api/mp_math/equation.md
2024-08-28 22:07:43 +08:00

3.7 KiB
Raw Blame History

title
mbcp.mp_math.equation

def get_partial_derivative_func(func: MultiVarsFunc = EPSILON) -> MultiVarsFunc

求N元函数一阶偏导函数。这玩意不太稳定慎用。

Args:

  • func: 函数

  • var: 变量位置,可为整数(一阶偏导)或整数元组(高阶偏导)

  • epsilon: 偏移量

Return:

  • 偏导函数

Raises:

  • ValueError 无效变量类型
Source code
def get_partial_derivative_func(func: MultiVarsFunc, var: int | tuple[int, ...], epsilon: Number=EPSILON) -> MultiVarsFunc:
    """
    求N元函数一阶偏导函数。这玩意不太稳定慎用。
    Args:
        func: 函数
        var: 变量位置,可为整数(一阶偏导)或整数元组(高阶偏导)
        epsilon: 偏移量
    Returns:
        偏导函数
    Raises:
        ValueError: 无效变量类型
    """
    if isinstance(var, int):

        def partial_derivative_func(*args: Var) -> Var:
            args_list_plus = list(args)
            args_list_plus[var] += epsilon
            args_list_minus = list(args)
            args_list_minus[var] -= epsilon
            return (func(*args_list_plus) - func(*args_list_minus)) / (2 * epsilon)
        return partial_derivative_func
    elif isinstance(var, tuple):

        def high_order_partial_derivative_func(*args: Var) -> Var:
            result_func = func
            for v in var:
                result_func = get_partial_derivative_func(result_func, v, epsilon)
            return result_func(*args)
        return high_order_partial_derivative_func
    else:
        raise ValueError('Invalid var type')

def partial_derivative_func() -> Var

Source code
def partial_derivative_func(*args: Var) -> Var:
    args_list_plus = list(args)
    args_list_plus[var] += epsilon
    args_list_minus = list(args)
    args_list_minus[var] -= epsilon
    return (func(*args_list_plus) - func(*args_list_minus)) / (2 * epsilon)

def high_order_partial_derivative_func() -> Var

Source code
def high_order_partial_derivative_func(*args: Var) -> Var:
    result_func = func
    for v in var:
        result_func = get_partial_derivative_func(result_func, v, epsilon)
    return result_func(*args)

class CurveEquation

def __init__(self, x_func: OneVarFunc, y_func: OneVarFunc, z_func: OneVarFunc)

曲线方程。

Args:

  • x_func: x函数

  • y_func: y函数

  • z_func: z函数

Source code
def __init__(self, x_func: OneVarFunc, y_func: OneVarFunc, z_func: OneVarFunc):
    """
        曲线方程。
        Args:
            x_func: x函数
            y_func: y函数
            z_func: z函数
        """
    self.x_func = x_func
    self.y_func = y_func
    self.z_func = z_func

def __call__(self) -> Point3 | tuple[Point3, ...]

计算曲线上的点。

Args:

  • *t:

  • 参数:

Source code
def __call__(self, *t: Var) -> Point3 | tuple[Point3, ...]:
    """
        计算曲线上的点。
        Args:
            *t:
                参数
        Returns:

        """
    if len(t) == 1:
        return Point3(self.x_func(t[0]), self.y_func(t[0]), self.z_func(t[0]))
    else:
        return tuple([Point3(x, y, z) for x, y, z in zip(self.x_func(t), self.y_func(t), self.z_func(t))])

def __str__(self)

Source code
def __str__(self):
    return 'CurveEquation()'

var result_func = get_partial_derivative_func(result_func, v, epsilon)