Добавить легенду для анимации (художников) в matplotlib
Я сделал анимацию из набора таких изображений (10 снимков):
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Circle
import time
infile = open ('out.txt')
frame_counter = 0
N_p = 100
N_step = 10
N_line = N_p*N_step
for s in xrange(N_step):
x, y = [], []
for i in xrange(N_p):
data = infile.readline()
raw = data.split()
x.append(float(raw[0]))
y.append(float(raw[1]))
xnp = np.array(x)
ynp = np.array(y)
fig = plt.figure(0)
ax = fig.add_subplot(111, aspect='equal')
for x, y in zip(xnp, ynp):
cir = Circle(xy = (x, y), radius = 1)
cir.set_facecolor('red')
ax.add_artist(cir)
cir.set_clip_box(ax.bbox)
ax.set_xlim(-10, 150)
ax.set_ylim(-10, 150)
fig.savefig("step.%04d.png" % frame_counter)
ax.remove()
frame_counter +=1
Теперь я хочу добавить легенду к каждому изображению, показывающему шаг по времени. Для этого я должен установить легенды для каждого из этих 10 изображений. Проблема в том, что я проверил разные вещи, какax.set_label , cir.set_label... и я получаю такие ошибки:
UserWarning: No labelled objects found. Use label='...' kwarg on individual plots
В соответствии с этой ошибкой я должен добавить метку к своим индивидуальным сюжетам, но так как это сюжет художников, я не знаю, как я могу это сделать.