P-2.2.0 用Python绘制函数图像

在学习数学时，经常希望绘制2/3D的函数图像/图像帮助理解
以下是实现方法：

2D图像:

from matplotlib import pyplot as plt  
import numpy as np  
from mpl_toolkits.mplot3d import axes3d

low=lambda x:10000 if x>10000 else -10000 if x<-10000 else x

f=lambda x:x**5#设计一个函数

start=-10 #输入需要绘制的起始值（从左到右）
stop=10 #输入需要绘制的终点值
step=0.01#输入步长


num=(stop-start)/step #计算点的个数
x = np.linspace(start,stop,num)
y = f(x)

for i in range(len(y)):#再应用一个low函数以防止函数值过大导致显示错误（可选）#若函数无法直接应用到np数组上可以使用for来逐个应用
    y[i]=low(y[i])
z=y

fig=plt.figure(figsize=(6,6))#建立一个对象并设置窗体的大小，使其为正方形，好看 #注意 可以建立多个对象，但plt指令只会对最后一个指定的对象进行操作（查看过源码了）

plt.plot(x, y,label='First Curve')#在当前的对象上进行操作