python matplotlib画图

python matplotlib画图

sin函数

import numpy as np
import matplotlib.pyplot as plt
import matplotlib
# 使用窗口
matplotlib.use("TKAgg")

# 创建图像
fig = plt.figure()
# x取值范围
# np.linspace参数, start开始位置, stop结束位置, num点数
x = np.linspace(-2 * np.pi, 2 * np.pi, 1000)
y = np.sin(x)
# 画图
plt.plot(x, y)
plt.show()

sin函数动画

from matplotlib import animation
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.use("TKAgg")

fig = plt.figure()
# x取值范围
# np.linspace参数, start开始位置, stop结束位置, num点数
x = np.linspace(-4 * np.pi, 4 * np.pi, 1000)

# 轴
ax = plt.axes(xlim=(np.pi * -2, np.pi * 2), ylim=(-1, 1))
line, = ax.plot(x, np.sin(x), color='blue')

# 初始化函数
def init():
    line.set_ydata(np.sin(x))
    return line,

# 每帧执行
def animate(i):
    # i从0开始
    line.set_ydata(np.sin(x + i/10))
    return line,

# func: 动画函数, init_func: 初始化函数, frames: 帧数, interval: 刷新间隔, blit: 刷新所有点
ani = animation.FuncAnimation(fig, func=animate, init_func=init,
                              frames=1000, interval=20, blit=True)
plt.show()