python - 在 python 中使用普通函数 f(x)

Question

我想在 python 中使用函数 f(x)。类似 ax^2 + bx + c(多项式)的东西。我想使用 for 循环和列表来做到这一点。这是我目前所拥有的:

def f(a,x):
    for i in range (0, len(a)):
        i = ([a]*x**i)
    print (i)

例如:当我填写 f([5,2,3],5) 时，我必须得到:3*5^0 + 2*5^1 + 5*5^2。有人知道我如何更改我的代码以便输出将是该多项式的结果吗？

Answer 1

使用 generator expression使用枚举

>>> sum(data*x**index for index, data in enumerate(reversed(a)))
138

你可以像这样在你的 func 中使用这个理解:

def f(a,x):
    print(sum(data*x**index for index, data in enumerate(reversed(a))))    
>>> f([5,2,3],5)    
138

已编辑:@Paul Panzer 建议的更优化版本