0

我正在创建一个包含多组数据的 3D 散点图，并为整个图形使用颜色图。代码如下所示：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

for R in [range(0,10), range(5,15), range(10,20)]:
  data = [np.array(R), np.array(range(10)), np.array(range(10))]
  AX = ax.scatter(*data, c=data[0], vmin=0, vmax=20, cmap=plt.cm.jet)
  def forceUpdate(event): AX.changed()
  fig.canvas.mpl_connect('draw_event', forceUpdate)

plt.colorbar(AX)

这很好用，但只要我保存它或旋转绘图，第一个和第二个散点上的颜色就会变成蓝色。

3D 散点图会丢失除最后一个数据集以外的所有颜色图信息。

强制更新通过保留颜色但仅在绘制的最后一个散点图上起作用。我尝试制作一个更新所有散点图的循环，但我得到与上面相同的结果：

AX = []
for R in [range(0,10), range(5,15), range(10,20)]:
  data = [np.array(R), np.array(range(10)), np.array(range(10))]
  AX.append(ax.scatter(*data, c=data[0], vmin=0, vmax=20, cmap=plt.cm.jet))
for i in AX:
  def forceUpdate(event): i.changed()
  fig.canvas.mpl_connect('draw_event', forceUpdate)

知道如何确保所有散点图都被更新以使颜色不会消失吗？

谢谢！

4

1 回答 1

0

修改了您的代码以使其执行任何操作：

>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> from mpl_toolkits.mplot3d import Axes3D
>>> AX = \[\]
>>> fig = plt.figure()
>>> ax = fig.add_subplot(111, projection='3d')
>>> for R in \[range(0,10), range(5,15), range(10,20)\]:
...   data = \[np.array(R), np.array(range(10)), np.array(range(10))\]
...   AX = ax.scatter(*data, c=data\[0\], vmin=0, vmax=20, cmap=plt.cm.jet)
...   def forceUpdate(event): AX.changed()
...   fig.canvas.mpl_connect('draw_event', forceUpdate)
... 
9
10
11
>>> plt.colorbar(AX)
<matplotlib.colorbar.Colorbar instance at 0x36265a8>
>>> plt.show()

然后我得到： 旋转后保存结果

所以上面的代码是有效的。如果您现有的代码不是，那么我建议您尝试上面的确切代码，如果这不起作用，请查看您正在使用的代码版本，如果它确实有效，那么您将不得不调查它与您的代码之间的差异实际代码（而不是您的示例代码）。

于 2013-08-14T18:10:29.320 回答