python - 重绘后 Matplotlib 3D 散点颜色丢失

Question

与this question相关，我想要一个 3D 散点图，每个点都有规定的颜色。问题中发布的示例在我的系统上有效，但是在第一次重绘之后(例如在保存之后或者如果我旋转图像)颜色似乎丢失了，即所有点都以蓝色绘制，具有通常的深度信息。请参阅下面的修改示例。

我的系统是 Python 2.6.7，在 mac 10.8.0 上从 macports 安装了 matplotlib 1.1.0。我使用 MacOSX 后端。

有谁知道如何规避这个问题？

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# Create Map
cm = plt.get_cmap("RdYlGn")

x = np.random.rand(30)
y = np.random.rand(30)
z = np.random.rand(30)

col = np.arange(30)

fig = plt.figure()
ax3D = fig.add_subplot(111, projection='3d')
ax3D.scatter(x, y, z, s=30, c=col, marker='o', cmap=cm)

plt.savefig('image1.png')
plt.savefig('image2.png')

这是两张图片，我得到:

Answer 1

目前还不清楚为什么会这样，这肯定是一个错误。在这里，我提供了一个 hack 来获得您想要的结果，尽管它不像人们想要的那样是自动的。

由于某些原因，表示散点的 Patch3DCollection 在第一次渲染后没有更新。这个更新是必不可少的，因为它是为每个收集补丁设置独特颜色的地方。要强制它重新初始化，您可以在 Patch3DCollection 上使用 changed 方法(实际上是一个 ScalarMappable 方法)，这只是记录了一个更改发生了。绘制图形时，它会检查是否发生更新，然后重新定义颜色。如果没有，则跳过此过程。

要强制此更新自动发生，人们希望在每个“绘制”事件上都执行此操作。为此，必须 register a method使用 canvas 的 mpl_connect 方法(参见链接教程)。

这个例子展示了两次保存图形如何保留颜色映射，但是如果你取消注释 plt.show() 行，它仍然有效(例如旋转)。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# Create Map
cm = plt.get_cmap("RdYlGn")

# added a seed so consistant plotting of points
np.random.seed(101)
x = np.random.rand(30)
y = np.random.rand(30)
z = np.random.rand(30)

col = np.arange(30)

fig = plt.figure()
#ax = fig.add_subplot(111)
#scatCollection = ax.scatter(x,y,
ax3D = fig.add_subplot(111, projection='3d')
# keep track of the Patch3DCollection:
scatCollection = ax3D.scatter(x, y, z, s=30, 
                            c=col, 
                            marker='o',
                            cmap=cm
                            )
def forceUpdate(event):
    global scatCollection
    scatCollection.changed()

fig.canvas.mpl_connect('draw_event',forceUpdate)

#plt.show()

plt.savefig('image1.png')

plt.savefig('image2.png')

理想情况下，不需要这样做，并且应该使用其他方法访问全局 scatCollection(我正在努力这样做)。但这现在有效......