[python]函数画图

python有丰富的数学库和绘图库，可以支持部分matlab功能，用于创建数学模型分析。

本文主要记录函数画图相关内容，供自己记录使用。

以下是调试程序

"""
画图
"""
import numpy as np   # numpy为数学库
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D


# 一阶函数图像
# 设置初始参数
x1 = np.random.randint(10, size=100)
x2 = np.linspace(0, 10, 1000)
x3 = np.arange(1, 10, 0.1)
y1 = 1/(1+0.5*x1)-(1-x1)
y2 = np.sin(x2)
y3 = np.exp(-(x3-5)**2/2*5)

fig = plt.figure("图1")
plt.plot(x1, y1)
fig = plt.figure()
plt.plot(x2, y2)
fig = plt.figure("图1")
plt.plot(x3, y3)



# 二阶函数图像
x = np.arange(1, 100, 1)
y = np.arange(1, 100, 1)
W, Q = np.meshgrid(x, y)  # 网格的创建，这个是关键
x = np.arange(-2 * np.pi, 2 * np.pi, 0.1)
y = np.arange(-2 * np.pi, 2 * np.pi, 0.1)
X, Y = np.meshgrid(x, y)  # 网格的创建，这个是关键
Z = np.sin(X) * np.cos(Y)

fig = plt.figure()
ax = Axes3D(fig)
plt.xlabel('x')
plt.ylabel('y')
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')

# 图像展示
plt.show()