38

我正在尝试制作一个函数来绘制我告诉它的任何公式。

import numpy as np  
import matplotlib.pyplot as plt  
def graph(formula, x_range):  
    x = np.array(x_range)  
    y = formula  
    plt.plot(x, y)  
    plt.show()

当我尝试调用它时，会发生以下错误，我相信它正在尝试在它到达之前进行乘法运算y = formula。

graph(x**3+2*x-4, range(-10, 11))

Traceback (most recent call last):  
  File "<pyshell#23>", line 1, in <module>  
    graph(x**3+2*x-4, range(-10, 11))  
NameError: name 'x' is not defined
4

3 回答 3

73

您的猜测是正确的：代码正在尝试x**3+2*x-4立即评估。不幸的是，您无法真正阻止它这样做。好消息是，在 Python 中，函数是一等对象，我的意思是您可以像对待任何其他变量一样对待它们。因此，要修复您的功能，我们可以这样做：

import numpy as np  
import matplotlib.pyplot as plt  

def graph(formula, x_range):  
    x = np.array(x_range)  
    y = formula(x)  # <- note now we're calling the function 'formula' with x
    plt.plot(x, y)  
    plt.show()  

def my_formula(x):
    return x**3+2*x-4

graph(my_formula, range(-10, 11))

如果您想在一行中完成所有操作，则可以使用所谓的lambda函数，它只是一个没有名称的简短函数，您不使用defor return：

graph(lambda x: x**3+2*x-4, range(-10, 11))

而不是range，您可以查看np.arange（允许非整数增量）和np.linspace，它允许您指定开始、停止和要使用的点数。

于 2012-12-22T06:32:16.840 回答
32

这是因为在行

graph(x**3+2*x-4, range(-10, 11))

x 未定义。

最简单的方法是将要绘制的函数作为字符串传递，并将eval其作为表达式求值。

所以你的代码修改最少

import numpy as np  
import matplotlib.pyplot as plt  
def graph(formula, x_range):  
    x = np.array(x_range)  
    y = eval(formula)
    plt.plot(x, y)  
    plt.show()

你可以把它称为

graph('x**3+2*x-4', range(-10, 11))
于 2012-12-22T06:37:51.860 回答
7

要绘制未针对特定变量（如圆或双曲线）求解的方程：

import numpy as np  
import matplotlib.pyplot as plt  
plt.figure() # Create a new figure window
xlist = np.linspace(-2.0, 2.0, 100) # Create 1-D arrays for x,y dimensions
ylist = np.linspace(-2.0, 2.0, 100) 
X,Y = np.meshgrid(xlist, ylist) # Create 2-D grid xlist,ylist values
F = X**2 + Y**2 - 1  #  'Circle Equation
plt.contour(X, Y, F, [0], colors = 'k', linestyles = 'solid')
plt.show()

更多关于它：http ://courses.csail.mit.edu/6.867/wiki/images/3/3f/Plot-python.pdf

于 2017-12-19T21:50:58.883 回答