Matplotlib 绘制一条螺旋曲线

使用Matplotlib绘制一条螺旋曲线,如下图:

Matplotlib 绘制一条螺旋曲线

跟前面的例子是一样的,采用回调函数的方式来实现,首先要导入动画库:

from matplotlib import animation

这样就可以使用animation的功能了,最重要的是使用animation.FuncAnimation和animation.FFMpegWriter函数。

接着定义重复时的清空函数:

def init():
global xdata, ydata
xdata, ydata = [], []
line.set_data([], [])
return line,

每次重复开始时,需要把全局变量xdata和ydata清空,否则绘图数据已经在变量里,就看不到绘图更新了,因为一次性就已经把整个图绘完了。这里调用set_data函数来清空绘图区域。



再定义动画更新函数:

def animate(i):

# t 帧序号
t = 0.1 * i

# x, y 绘图
x = t * np.sin(t)
y = t * np.cos(t)

# 增加到列表,连接起所有点。
xdata.append(x)
ydata.append(y)
line.set_data(xdata, ydata)

return line,

在这段代码里,通过帧号的改变来计算参数t,而参数t用来计算圆的坐标,再把这个坐标添加全局变量里保存,最后根据所有坐标点进行绘图,就可以不断地更新图案。

anim = animation.FuncAnimation(fig, animate, init_func = init,

frames = 500, interval = 20, blit = True)

这行代码是设置动画回调函数,fig是绘图区域,animate是回调更新函数,init是重复时调用的函数,frames是500帧长度,interval是间隔20毫秒更新一帧。

把前面的代码进行综合,就可以构造成完整的例子:

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation

plt.rcParams['animation.ffmpeg_path'] = r'D:\ffmpeg-2020-12-20\bin\ffmpeg.exe'


# 创建绘图轴对象,创建一个空白窗口
fig = plt.figure() 
axis = plt.axes(xlim =(-50, 50), ylim =(-50, 50)) 
line, = axis.plot([], [], lw = 2) 
#创建全局数据保存列表
xdata, ydata = [], []
#定义每次重复清空函数
def init():
    global xdata, ydata
    xdata, ydata = [], []
    line.set_data([], []) 
    return line,
#每帧动画更新函数
def animate(i): 
    # t 帧序号
    t = 0.1 * i 

    # x, y 绘图 
    x = t * np.sin(t) 
    y = t * np.cos(t) 

    # 增加到列表,连接起所有点。
    xdata.append(x) 
    ydata.append(y) 
    line.set_data(xdata, ydata) 

    return line, 

anim = animation.FuncAnimation(fig, animate, init_func = init,
                               frames = 500, interval = 20, blit = True) 

#保存的动画视频文件名为当前文件夹下的ani-055.mp4,帧率为20帧每秒,格式为MP4。
FFwriter = animation.FFMpegWriter(fps=20, extra_args=['-vcodec', 'libx264'],
                                  metadata = {'title':'demo', 'artist':'蔡军生'})
anim.save('ani-055.mp4', writer = FFwriter)

plt.show()  # 会一直循环播放动画

当你运行完成时,就可以到源码所在的目录找到ani-055.mp4文件,点击播放就可以看到整个动画过程的视频了。

赞(0)
未经允许不得转载:极客笔记 » Matplotlib 绘制一条螺旋曲线

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址
Matplotlib 入门
Matplotlib 入门Matplotlib 安装之Linux操作系统Matplotlib 安装之Windows系统Matplotlib 安装之Mac OS X系统Matplotlib 安装之使用 Windows Installer安装程序Matplotlib 使用*.whl文件快速安装Matplotlib 绘图区域Matplotlib 设置绘图区域大小Matplotlib 网格线的颜色和刻度Matplotlib 坐标轴的标签和标题Matplotlib 指南Matplotlib pyplot指南
Matplotlib 几何图形
Matplotlib 画直线Matplotlib 图形失真Matplotlib 画圆弧示意图Matplotlib 用直线方式画圆Matplotlib 画椭圆示意图Matplotlib 画旋转向量示意图Matplotlib 使用patches绘制几何图形Matplotlib 使用模块patches的Wedge实现饼图Matplotlib 嵌套子图显示Matplotlib 两个子图显示
Matplotlib 颜色
Matplotlib 颜色Matplotlib 多边形颜色填充Matplotlib 不规则颜色填充Matplotlib 条件选择区域颜色填充
Matplotlib 坐标轴
Matplotlib 辅助指示坐标轴Matplotlib 公式显示Matplotlib 显示数学曲线Matplotlib 同一坐标里显示多条曲线Matplotlib 同一坐标里双Y轴显示Matplotlib 双Y轴显示时把图例显示到一起Matplotlib 三个Y轴显示Matplotlib 隐藏刻度方式显示三个Y轴Matplotlib 四个Y轴显示Matplotlib 一元线性回归显示Matplotlib X轴日期显示Matplotlib X轴年份采用四位显示Matplotlib X轴按小时显示Matplotlib X轴设置为弧度坐标Matplotlib X轴显示角度Matplotlib X轴显示特殊的标签Matplotlib 移动坐标轴位置Matplotlib 与数据区图像相对位置移动坐标轴位置
Matplotlib 函数
Matplotlib 函数plot 展现变量的趋势变化Matplotlib 函数plot 创建带文本标签的折线图Matplotlib 函数scatter 寻找变量之间的关系Matplotlib 函数xlim 设置x轴的数值显示范围Matplotlib 函数xlabel 设置x轴的标签文本Matplotlib 函数grid 绘制刻度线的网格线Matplotlib 函数axhline 绘制平行于x轴的水平参考线Matplotlib 函数axvspan 绘制垂直于x轴的参考区域Matplotlib 函数annotate 添加图形内容细节的指向型注释文本Matplotlib 函数text 添加图形内容细节的无指向型注释文本Matplotlib 函数title 添加图形内容的标题Matplotlib 函数legend 标示不同图形的文本标签图例Matplotlib 函数bar 用于绘制柱状图Matplotlib 函数barh 用于绘制条形图Matplotlib 函数hist 用于绘制直方图Matplotlib 函数pie 用于绘制饼图Matplotlib 函数polar 用于绘制极线图Matplotlib 函数scatter 用于绘制气泡图Matplotlib 函数stem 用于绘制棉棒图Matplotlib 函数boxplot 用于绘制箱线图Matplotlib 函数errorbar 用于绘制误差棒图Matplotlib 函数subplots和函数subplot区别
Matplotlib 动画和视频
Matplotlib ffmpeg安装Matplotlib 基本动画过程Matplotlib FuncAnimation方式创建动画Matplotlib ArtistAnimation方式创建动画Matplotlib 绘制一条螺旋曲线Matplotlib 输出gif文件Matplotlib 调用ImageMagick输出gif文件
Matplotlib 3D
Matplotlib 3D的坐标系Matplotlib 绘制3D图像
Matplotlib 导入导出
Matplotlib 保存绘图结果为PDF文件
Matplotlib 画线
Matplotlib 画线