如何使用Python和matplotlib绘制任意数学函数

Question

我想绘制任何数学函数，如y=x**2或y=sin(x(/x(基本上任何东西(。问题是，我只能绘制numpy中定义的函数，如np.sin、np.cos或np.tan。这里的全部要点是如何避免"；硬编码"；向量，然后绘制这些向量。

如何绘制任意函数？以下是我目前所拥有的：

import matplotlib.pyplot as plt
import numpy as np
#Let a be starting point on x
#Let b be end point on x
#Let f be the function you wish to plot
CornflowerBlue="6495ed"
def plotf(a,b,f):

x=np.linspace(a,b,num=b*10)
y=np.array([])

for i in range (len(x)):

y = np.append(y,f(x[i]))

plt.grid(True)
plt.plot(x,y,color="CornflowerBlue")
plt.axhline(y=0,color="black")
return 

Answer 1

您可以尝试sympy，因为它应该允许您在不显式硬编码的情况下处理各种数学函数：

from sympy import *
def plotf(a,b,f):
plot(f, xlim = (a,b)) 
plotf(-10, 10, "sin(x)/x") # or plotf(0, 2, "x**2"), or (almost) whatever

有关详细信息，请参阅https://docs.sympy.org/latest/modules/plotting.html！