Python：3D散点丢失颜色图

Question

我正在创建一个包含多组数据的 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)

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

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

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)

知道如何确保所有散点图都在更新，这样颜色就不会消失吗？

谢谢！

Answer 1

修改了你的代码，让它可以做任何事情：

>>> 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()

然后我得到：

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