Matplotlib：如何从轴上仅删除一个轮廓元素与其他绘制元素？

Question

我正在尝试对高斯混合（高斯混合模型）的均值和协方差的估计进行动画处理，我需要在每次迭代时更新均值和协方差图。

重绘方法非常简单，因为我使用具有set_data 方法的行，我可以在每次更新时调用该方法。不幸的是，更新协方差是另一回事，因为 contour 元素表示为 QuadContourSet 对象并且没有 set_data 方法。

这是一个玩具示例：

import numpy as np
from matplotlib import mlab

# Toy data points (these are constant)
plt.plot(np.arange(-3,3,0.1), np.arange(-3,3,0.1))

x = np.arange(-5.0, 5.0, 0.1)
y = np.arange(-5.0, 5.0, 0.1)
X, Y = np.meshgrid(x, y)

# First toy iteration
Z1 = mlab.bivariate_normal(X, Y, 
                           1, 1, 
                           0, 0)

covariance1 = plt.contour(X, Y, Z1)

# Second toy iteration
Z2 = mlab.bivariate_normal(X, Y, 
                       1, 1, 
                       0, 3)

covariance2 = plt.contour(X, Y, Z2)

在实际问题中，我绘制均值、方差和数据点，我不想清除整个轴。

问题是如何在不删除其他元素的情况下删除第一个轮廓covariance1？

Answer 1

for coll in covariance1.collections:
    coll.remove()

然后更新。