获取轴 matplotlib 的所有艺术家?

Get all artists of an axes matplotlib?

我正在尝试在 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()

也许您正在寻找的是 ax.get_children(),但它 returns 一切(线条、线条集合、书脊、图例可能...),因此您必须以某种方式过滤结果.

如果你事先知道你正在绘制什么样的数据,这里有几个 Matplotlib 分隔艺术家的位置。

  • ax.lines 存储使用 ax.plot 创建的行。
  • ax.collections 存储使用 ax.add_collectionax.plot_surfaceax.contourax.contourf、...[ 创建的集合(线集合、多边形集合) =35=]
  • ax.images 存储使用 ax.imshow 创建的艺术家。
  • ax.patches 包含使用 ax.barax.fill、....
  • 创建的艺术家
  • ax.tables 包含使用 ax.table.
  • 创建的艺术家