Question

我有一个动画python图。虽然每次计数器可以除以100时我都可以添加注释（箭头），但我无法弄清楚如何将箭头（或将其附加）添加到图形中（请参阅我的屏幕截图））。目前，add_annotation只能工作1秒，然后消失（它应该至少再持续10秒左右）。

转述了从here

取得的例子

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

fig = plt.figure()
ax1 = fig.add_subplot(1,1,1)

z = np.random.normal(0,1,255)
u = 0.1
sd = 0.3
counter = 0
price = [100]
t = [0]

def add_annotation(annotated_message,value):
    plt.annotate(annotated_message, 
                 xy = (counter, value), xytext = (counter-5, value+20),
                 textcoords = 'offset points', ha = 'right', va = 'bottom',
                 bbox = dict(boxstyle = 'round,pad=0.1', fc = 'yellow', alpha = 0.5),
                 arrowprops = dict(arrowstyle = '->', connectionstyle = 'arc3,rad=0'))

def getNewPrice(s,mean,stdev):
    r = np.random.normal(0,1,1)
    priceToday = s + s*(mean/255+stdev/np.sqrt(225)*r)
    return priceToday


def animate(i):
    global t,u,sd,counter
    x = t
    y = price
    counter += 1
    x.append(counter)
    value = getNewPrice(price[counter-1],u,sd)
    y.append(value)
    ax1.clear()

    if counter%100==0:
        print "ping"
        add_annotation("100",value)

    plt.plot(x,y,color="blue")


ani = animation.FuncAnimation(fig,animate,interval=20)
plt.show()

Answer 1

您的代码存在一些问题，主要问题是您在动画的每一步（ax1.clear()）都清除了轴，因此您正在删除注释。

一种选择是使用所有注释保持数组，并且每次都重新创建它们。这显然不是最优雅的解决方案。

制作动画的正确方法是创建Line2D对象，然后使用Line2D.set_data()或Line2D.set_xdata()和Line2D.set_ydata()来更新线条绘制的坐标

请注意，动画功能应返回已修改的艺术家列表（至少如果您想使用blitting，这会提高性能）。因此，在创建注释时，需要返回Annotation对象，并将它们保存在列表中，并通过动画函数返回所有艺术家。

这很快被抛到一起，但应该给你一个起点：

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

fig = plt.figure()
ax1 = fig.add_subplot(1,1,1)

ax1.set_xlim((0,500))
ax1.set_ylim((0,200))

z = np.random.normal(0,1,255)
u = 0.1
sd = 0.3
counter = 0
price = [100]
t = [0]
artists = []
l, = plt.plot([],[],color="blue")


def add_annotation(annotated_message,value):
    annotation = plt.annotate(annotated_message, 
                 xy = (counter, value), xytext = (counter-5, value+20),
                 textcoords = 'offset points', ha = 'right', va = 'bottom',
                 bbox = dict(boxstyle = 'round,pad=0.1', fc = 'yellow', alpha = 0.5),
                 arrowprops = dict(arrowstyle = '->', connectionstyle = 'arc3,rad=0'))
    return annotation


def getNewPrice(s,mean,stdev):
    r = np.random.normal(0,1,1)
    priceToday = s + s*(mean/255+stdev/np.sqrt(225)*r)
    return priceToday


def animate(i):
    global t,u,sd,counter,artists
    x = t
    y = price
    counter += 1
    x.append(counter)
    value = getNewPrice(price[counter-1],u,sd)
    y.append(value)
    l.set_data(x,y)

    if counter%100==0:
        print(artists)
        new_annotation = add_annotation("100",value)
        artists.append(new_annotation)

    return [l]+artists


ani = animation.FuncAnimation(fig,animate,interval=20,frames=500)

使用matplotlib的动画图形 - 无法维持/附加注释

1 个答案: