python – Matplotlib FuncAnimation慢

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – Matplotlib FuncAnimation慢，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含3072字，纯文字阅读大概需要5分钟。

内容图文

我使用matplotlib的FuncAnimation创建了一个表面绑定过程的简单动画.结果却非常缓慢.我怀疑是因为我正在重绘每一帧的所有元素,但我还没有找到解决办法.任何帮助赞赏.

import matplotlib
matplotlib.use('TKAgg')      # import proper graphics back-end for Mac OS X


import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation
from matplotlib.collections import PatchCollection
from random import *

nx = 20     # x size of lattice
ny = 20     # y size of lattice

pAds = 0.01     # adsorption probability per time step
pDes = 0.0075   # desorption probability per time step

tMax = 500     # number of time steps

surface = np.zeros((nx,ny))              # create surface
xc = [0]
yc = [0]


# initialization and time step of simulation

def init():
    # initialize an empty list of circles
    patches = []            # empty array to hold drawable objects
    for x in range(0,nx):
        for y in range(0,ny):
            if(surface[x][y] == 0):
                patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='w')))
    lines, = ax_covr.plot([],[])
    patches.append(lines)
    return patches

def animate(i): 
    patches = []            # empty array of circles to be drawn
    for x in range(0,nx):
        for y in range(0,ny):
            if(surface[x][y] == 0):
                if(random() < pAds):
                    surface[x][y] = 1
                    patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='b')))
                else:
                    patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='w')))
            else:
                if(random()<pDes):
                    surface[x][y] = 0
                    patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='w')))
                else:
                    patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='b')))
    coverage = np.sum(surface)/(nx*ny)
    xc.append(i)
    yc.append(coverage)
    lines, = ax_covr.plot(xc,yc,'ro',ms=2,lw=0)
    patches.append(lines)
    return patches

# set up figure and animate


fig = plt.figure()
ax_surf = plt.subplot2grid((1, 2), (0, 0))
ax_covr = plt.subplot2grid((1, 2), (0, 1))
ax_surf.set_xlim(0,nx)
ax_surf.set_ylim(0,ny)
ax_covr.set_xlim(0,tMax)
ax_covr.set_ylim(0,1)

ax_surf.set_aspect(1)
ax_surf.axis('off')

ax_covr.set_aspect(tMax)

ax_surf.hold(False)

anim = animation.FuncAnimation(fig, animate, init_func=init, frames=tMax, interval=0, blit=True,repeat=False)
plt.show()

解决方法:

使用先前init / animate调用的补丁而不是每次都创建新补丁.

以下是代码的修改版本.

patches = []

def init():
    global patches
    if patches:
        # prevent the second call of the init()
        return patches
    # initialize an empty list of circles
    for x in range(nx):
        for y in range(ny):
            if(surface[x][y] == 0):
                patches.append(ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='w')))
    lines, = ax_covr.plot([],[])
    patches.append(lines)
    return patches

def animate(i): 
    global patches
    idx = 0
    for x in range(nx):
        for y in range(ny):
            if surface[x][y] == 0:
                if random() < pAds:
                    surface[x][y] = 1
                    patches[idx] = ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='b'))
            else:
                if(random()<pDes):
                    surface[x][y] = 0
                    patches[idx] = ax_surf.add_patch(plt.Circle((x+0.5,y+0.5),0.45,color='w'))
            idx += 1
    coverage = np.sum(surface)/(nx*ny)
    xc.append(i)
    yc.append(coverage)
    lines, = ax_covr.plot(xc,yc,'ro',ms=2,lw=0)
    patches[idx] = lines
    return patches

注意：使用全局变量来最小化修改.

内容总结

以上是互联网集市为您收集整理的python – Matplotlib FuncAnimation慢全部内容，希望文章能够帮你解决python – Matplotlib FuncAnimation慢所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端