Defining and evaluating a function of decreasing powers

userGAJ21 0
Viewed 41

I would like to write a function and input only two parameters: the degree of the function as an integer and the parameters as a list, and be able to evaluate it. The output of the function should be evaluated by using another method.

def func(degree: int, params: list) -> Object
    pass

In: f = func( 3, [3, 2, -1, 8] )
In:f
Out: 3*x**3 + 2*x**2 - 1*x + 8

In:f(3)
Out: 104

Currently, I'm doing manually and it's tedious for higher-order function:

def func(t,a,b,c,d):
    return a*pow(t,3)+b*pow(t,2)+c*t+d
sympy math function python
4 Answers

All you need to do is send the list of coefficients to your function as ... a list.

def evaluate_polynomial(x : int, a : list):
    result = 0
    n = len(a)
    for i in range(n):
        result += a[i] * x**(n-i-1)
    return result

assert evaluate_polynomial(3, [3, 2, -1, 8]) == 104
assert evaluate_polynomial(3, [-1, 4, -3, 1, -4, 3, 2, -1, 8]) == 23
r3mainer0

SymPy's Poly pretty much has all the batteries you need:

enter image description here

smichr0

A class will work best for you. It will help you to define a function, then evaluate the function using a variable. Check this implementation:

class CreateFunction:
    def __init__(self, degree: int, weights: list):
        self.degree = degree
        self.weights = weights
        self.powers = [i for i in range(self.degree,-1,-1)]

        # making powers of extreme right variables 0
        # if the list containing weights has more weights
        # than the list containing powers (exponents)
        if len(self.weights)>len(self.powers):
            difference = len(self.weights) - len(self.powers)
            additional_powers = [0 for i in range(difference)]
            self.powers += additional_powers


    def evaluate(self, variable: int):
        sum=0
        for (w, p) in zip(self.weights, self.powers):
            sum += w*variable**p
        return sum

func = CreateFunction(3, [3, 2, -1, 8])
out1 = func.evaluate(3)
print(f'out1={out1}')

out2 = func.evaluate(5)
print(f'out2={out2}')

func2 = CreateFunction(4, [3, 2, -1, 8, 17])
out3 = func2.evaluate(7)
print(f'out3={out3}')

output

out1=104
out2=428
out3=7913
Dalmas Otieno0

A build-in approach. No need to pass the degree as parameter, it will be the length of the list of coefficients. The coefs are ordered from highest to lower degree.

def polynomial(coefs):
    return lambda x: sum(c * x**(index-1) for c, index in  zip(coefs, range(len(coefs), -1, -1)))

p = polynomial([3, 2, -1, 8])

print(p(3))
#104
cards0
Related