python - 获取轴 matplotlib 的所有艺术家？

Question

我正在尝试在 mpl 中制作动画图，我决定最聪明的方法是使用 Gridspec 动态添加多个轴。我提供了一些我希望 Gridspec 在我的 Canvas 上制作轴的列和行，然后我可以用艺术家填充这些轴。

由于轴的数量是任意的，因此艺术家的数量是任意的，我认为最好像这样访问艺术家:

import matplotlib.pyplot as plt

fig = plt.figure(constrained_layout=False, figsize=(8,8), facecolor=bg_colour)
gs = fig.add_gridspec(nrows=nrows, ncols=ncols, hspace=0.1, wspace=0.1)

for i in range(nrows):
    for j in range(ncols):
        ax = plt.subplot(gs[i,j])

print(fig.axes[0].artists)

所以我的想法是，我可以动态创建我所有的轴，然后我还可以通过使用适当的索引切片 fig.axes 来动态地将艺术家添加到这些轴。一旦我选择了我的轴，我就可以做

fig.axes[0].plot([],[])

将空艺术家添加到右轴，然后我可以使用我的代码中的更新函数为其设置动画。

问题是无论我如何设置图形、轴和艺术家的创建，fig.axes[index].artists 列表始终为空。我不明白我怎么能发出为我的轴绘图的命令，然后不让艺术家出现在所有艺术家的列表中。 .artists 实际上不是我需要使用的所有艺术家的容器吗？我要查找的东西是否存放在其他地方？难道 mpl 一开始就不允许这种事情发生吗？

完整代码

import numpy as np
import itertools
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from matplotlib.ticker import AutoMinorLocator, MaxNLocator




def circle_points_x(period):

    t = np.linspace(0,2*np.pi,500)
    x = np.cos(period*t +np.pi/2)

    return x

def circle_points_y(period):

    t = np.linspace(0,2*np.pi,500)
    y = np.sin(period*t +np.pi/2)

    return y


# true distribution
ncols,nrows = (5,5)

bg_colour = np.array((84, 153, 215))/256
bg_colour = np.append(bg_colour,0.5)

fig = plt.figure(constrained_layout=False, figsize=(8,8), facecolor=bg_colour)

gs = fig.add_gridspec(nrows=nrows, ncols=ncols, hspace=0.1, wspace=0.1)


for i in range(nrows):
    for j in range(ncols):
        ax = plt.subplot(gs[i,j])
        ax.axis('off')


def init():

    for i in range(nrows):
        for j in range(ncols):
            x = circle_points_x(j+1)
            y = circle_points_y(i+1)

            fig.axes[i*nrows +j].plot(x,y,lw=0.5,aa=True)
            fig.axes[i*nrows +j].scatter([x[0]],[y[0]])


    print(fig.axes[0].artists)

    return list(itertools.chain(*[i.artists for i in fig.axes]))


def update(frame):

    for i in range(nrows):
        for j in range(ncols):
            # fig.axes[i*nrows +j].artists[0] # TURN THIS ON TO SEE THE ERROR
            pass
    return list(itertools.chain(*[i.artists for i in fig.axes]))




ani = FuncAnimation(fig, update, frames=range(1,8),repeat=False,init_func=init, blit=True,interval=40)
# ani.save('histogram.gif', dpi=300 ,writer="ffmpeg")
plt.show()

Answer 1

也许您正在寻找的是 ax.get_children()，但它会返回所有内容(行、行集合、书脊、图例可能...)，因此您必须过滤结果不知何故。

如果您事先知道您正在绘制哪种数据，这里有几个 Matplotlib 将艺术家分开的位置。

• ax.lines 存储用 ax.plot 创建的线。
• ax.collections 存储用ax.add_collection、ax.plot_surface、创建的集合(line collections、poly collections)斧.contour, ax.contourf, ...
• ax.images 存储使用 ax.imshow 创建的艺术家。
• ax.patches 包含使用 ax.bar、ax.fill、....
创建的艺术家
• ax.tables 包含使用 ax.table 创建的艺术家。