注释箭头道具中间的动画列表

Question

我的目标是在注释函数中间为df 设置动画。我可以让箭头进行动画处理，df 的第一个值会出现，但不会使用更新的坐标进行动画处理。为此，我将label.set_text 更改为(Number[i+1])，但这只是在第一帧的正确位置显示数字。由于未调用新坐标，因此位置不会更新。我尝试运行此代码来更新坐标，但它没有显示任何内容？

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
import random
from functools import partial
import pandas as pd

one_sample = partial(random.sample, range(100), 10)
a_data = [one_sample() for _ in range(1000)]
b_data = [one_sample() for _ in range(1000)]

df = pd.DataFrame(np.random.randint(0,100,size=(100, 1)), columns=list('A'))

fig, ax = plt.subplots(figsize = (8,6))
ax.set_xlim(0,100)
ax.set_ylim(0,100)

arrow = ax.annotate('', xy = (a_data[0][0], b_data[0][0]), xytext = (a_data[0][1],b_data[0][1]), arrowprops = {'arrowstyle': "<->", 'color':'black'}, ha = 'center')

Number = df[A']
label = plt.text(a_data[0][0], b_data[0][0], Number, fontsize = 8, ha = 'center')

def animate(i) :
    arrow_start = (a_data[0+i][0], b_data[0+i][0]) 
    arrow_end = (a_data[0+i][1], b_data[0+i][1])
    arrow.set_position(arrow_start)
    arrow.xy = arrow_end
    label.set_text(a_data[0+i][0], b_data[0+i][0])

ani = animation.FuncAnimation(fig, animate,
                         interval = 500, blit = False)

plt.draw()

Answer 1

虽然您可以使用plt.text 来显示标签，但您并不需要它。 ax.annotate 可以生成标签和箭头。可以指定标签字符串作为ax.annotate的第一个参数，

arrow = ax.annotate(Number[0], xy=(a_data[0][0], b_data[0][0]), ...

您可以通过调用arrow.set_text来更改标签：

arrow.set_text(Number[i])

---

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
import random
from functools import partial
import pandas as pd

one_sample = partial(random.sample, range(100), 10)
a_data = [one_sample() for _ in range(1000)]
b_data = [one_sample() for _ in range(1000)]

df = pd.DataFrame(np.random.randint(0, 100, size=(100, 1)), columns=list('A'))

fig, ax = plt.subplots(figsize=(8, 6))
ax.set_xlim(0, 100)
ax.set_ylim(0, 100)

Number = df['A']
arrow = ax.annotate(Number[0], xy=(a_data[0][0], b_data[0][0]), 
                    xytext=(a_data[0][1], b_data[0][1]), 
                    arrowprops={'arrowstyle': "<->", 'color': 'black'}, ha='center')

def animate(i):
    arrow_start = (a_data[0 + i][0], b_data[0 + i][0])
    arrow_end = (a_data[0 + i][1], b_data[0 + i][1])
    arrow.set_position(arrow_start)
    arrow.xy = arrow_end
    arrow.set_text(Number[i])
    return [arrow]

ani = animation.FuncAnimation(fig, animate, interval=500, blit=True)
plt.show()

---

要将标签放在箭头的中间，我相信您需要使用plt.text（或再次调用ax.annotate）。移动plt.text生成的标签，调用label.set_position：

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
import random
import math
from functools import partial
import pandas as pd

one_sample = partial(random.sample, range(100), 10)
a_data = [one_sample() for _ in range(1000)]
b_data = [one_sample() for _ in range(1000)]
df = pd.DataFrame(np.random.randint(0, 100, size=(100, 1)), columns=list('A'))
Number = df['A']

data = np.stack([a_data, b_data], axis=2)
# a_data and b_data contain more data than we are actually using, 
# so let's crop `data` to make the following code simpler:
data = data[:, :2, :]
middle = data.mean(axis=1)
# find the direction perpendicular to the arrow
perp_dir = (data[:, 0] - data[:, 1]).astype('float')
perp_dir = np.array((-perp_dir[:, 1], perp_dir[:, 0]))
perp_dir /= np.sqrt((perp_dir**2).sum(axis=0))
perp_dir = perp_dir.T
# shift middle by a little bit in the perpendicular direction
offset = 3.0
middle += offset * perp_dir

fig, ax = plt.subplots(figsize=(8, 6))
ax.set_xlim(0, 100)
ax.set_ylim(0, 100)

arrow = ax.annotate('', xy=data[0, 0], 
                    xytext=data[0, 1], 
                    arrowprops={'arrowstyle': "<->", 'color': 'black'}, 
                    ha='center')

label = plt.text(middle[0, 0], middle[0, 1], Number[0], fontsize = 8, 
                 ha = 'center')

def animate(i):
    arrow_start = data[i, 0]
    arrow_end = data[i, 1]
    arrow.set_position(arrow_start)
    arrow.xy = arrow_end
    label.set_text(Number[i])
    label.set_position(middle[i])
    return [arrow, label]

ani = animation.FuncAnimation(fig, animate, interval=500, blit=True)
plt.show()