Function _makeNumData

SLiCAP/SLiCAPmath.py:830–862

Returns a list of values y, where y[i] = yFunc(x[i]). :param yFunc: Function :type yFunc: sympy.Expr :param xVar: Variable that needs to be substituted in *yFunc* :type xVar: sympy.Symbol :param x: List with values of x :type x: list :param normalize: True if rat

(yFunc, xVar, x, normalize=True)

    return (mrgns, freqs)

def _makeNumData(yFunc, xVar, x, normalize=True):
    """
    Returns a list of values y, where y[i] = yFunc(x[i]).

    :param yFunc: Function
    :type yFunc: sympy.Expr

    :param xVar: Variable that needs to be substituted in *yFunc*
    :type xVar: sympy.Symbol

    :param x: List with values of x
    :type x: list

    :param normalize: True if rational function needs to be normalized. Defaults to True.
    :type normalize: Bool

    :return: list with y values: y[i] = yFunc(x[i]).
    :rtype:  list
    """
    if normalize:
        yFunc = normalizeRational(sp.N(yFunc), xVar)
    else:
        yFunc = sp.N(yFunc)
    if xVar in yFunc.atoms(sp.Symbol):
        # Check for Heaviside functions (not implemented in sp.lambdify)
        if len(yFunc.atoms(sp.Heaviside)) != 0:
            y = [sp.N(yFunc.xreplace({xVar: x[i]})).doit() for i in range(len(x))]
        else:
            func = sp.lambdify(xVar, yFunc, ini.lambdify)
            y = func(x)
    else:
        y = [sp.N(yFunc) for i in range(len(x))]
    return y

def _magFunc_f(LaplaceExpr, f):
    """