与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')
以下是两张图片,我得到:
答案 0 :(得分:11)
目前尚不清楚为什么会这样,而且肯定是一个错误。在这里,我提供了一个hack来获得你想要的结果,虽然它不像人们想要的那样 as 。
出于某种原因,表示散点的Patch3DCollection
在第一次渲染后不会更新。此更新至关重要,因为它是为每个集合修补程序设置唯一颜色的位置。要强制它重新初始化,您可以使用changed
上的Patch3DCollection
方法(实际上是ScalarMappable
方法),这只是记录更改发生的。绘制图形时,它会检查是否发生了更新,然后重新定义颜色。如果没有,则跳过此过程。
要强制此更新自动发生,我们希望在每个'draw'事件上执行此操作。要执行此操作,必须register a method使用canvas
的{{1}}方法(请参阅链接教程)。
此示例显示如何保存图形两次保留颜色映射,但如果取消注释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')
(我正在努力这样做)。但这现在有效......