我有一个功能
f(x) = sin(x/5.0)*exp(x/10.0) + 5*exp(-x/2.0)
我需要解线性方程组
w0 + w1x1 + w2(x1)**2 + ... + wn(x1)**n = f(x1)
我解决了，但我在绘制它时有问题
from math import sin, exp
from scipy import linalg
import numpy as np
b = []
def f(x):
return sin(x/5.0)*exp(x/10.0) + 5*exp(-x/2.0)
for i in [1, 15]:
b.append(f(i))
A = []
for i in [1, 15]:
ij = []
x0 = i ** 0
x1 = i ** 1
ij.append(x0)
ij.append(x1)
A.append(ij)
matrix = np.array(A)
b = np.array(b).T
x = linalg.solve(matrix, b)
from matplotlib import pyplot as plt
plt.plot(x, f(x))
但它返回
TypeError: only length-1 arrays can be converted to Python scalars
我怎么解决这个问题？
math.sin并math.exp期望标量输入。如果你传递一个数组，你会得到一个TypeError
In [34]: x
Out[34]: array([ 3.43914511, -0.18692825])
In [35]: math.sin(x)
TypeError: only length-1 arrays can be converted to Python scalars
from math import sin, exp从模块加载sin和并将它们定义为全局命名空间中的函数。调用NumPy 数组的函数版本也是如此：expmathf(x)mathsinx
def f(x):
return sin(x/5.0)*exp(x/10.0) + 5*exp(-x/2.0)
要修复错误，请改用 NumPysin和exp函数。
import numpy as np
def f(x):
return np.sin(x/5.0)*np.exp(x/10.0) + 5*np.exp(-x/2.0)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句