如何使用matplotlib动态更改scatter3d动画中的点颜色？

Question

我正在尝试使用matplotlib绘制几个3DScatter的动画。我成功地绘制了所有的点，但我正在努力克服颜色。即使我正在调用函数set_color(..)，也没有任何改变。 以下是我目前正在做的事情，to_plot是一个大小为total且(5120, 3)个浮点元素的数组，而colors是一个大小为total的数组{ {1}}个元素（等于'r'或'b'）：

(5120,)

Answer 1

散点图是Path3DCollection。它可以有一个与之关联的色彩映射，使其点根据颜色数组着色。

因此，您可以通过scat3D.set_array(colors[i])向colors[i] = [0,1,0,...,1,0,1]提供散点图列表。"bwr"。然后根据使用的色彩映射映射这些值。对于蓝/红色这很简单，因为已经存在从蓝色到红色的色彩映射import matplotlib.pyplot as plt import numpy as np import matplotlib.animation as animation from mpl_toolkits.mplot3d import Axes3D import matplotlib.patches as mpatches total = 10 num_whatever = 100 to_plot = [np.random.rand(num_whatever, 3) for i in range(total)] colors = [np.tile([0,1],num_whatever//2) for i in range(total)] red_patch = mpatches.Patch(color='red', label='Men') blue_patch = mpatches.Patch(color='blue', label='Women') fig = plt.figure() ax3d = Axes3D(fig) scat3D = ax3d.scatter([],[],[], s=10, cmap="bwr", vmin=0, vmax=1) scat3D.set_cmap("bwr") # cmap argument above is ignored, so set it manually ttl = ax3d.text2D(0.05, 0.95, "", transform=ax3d.transAxes) def update_plot(i): print i, to_plot[i].shape ttl.set_text('PCA on 3 components at step = {}'.format(i*20)) scat3D._offsets3d = np.transpose(to_plot[i]) scat3D.set_array(colors[i]) return scat3D, def init(): scat3D.set_offsets([[],[],[]]) plt.style.use('ggplot') plt.legend(handles=[red_patch, blue_patch]) ani = animation.FuncAnimation(fig, update_plot, init_func=init, blit=False, interval=100, frames=xrange(total)) ani.save("ani.gif", writer="imagemagick") plt.show() 。

{{1}}

Answer 2

调用set_color失败的原因如下：https://github.com/matplotlib/matplotlib/issues/13035

  

...该错误是由于set_facecolor未设置_facecolor3d所致，因为它是从基类（Collection）设置_facecolors继承而来的。边缘颜色也一样。

是的，这是matplotlib中的错误。

因此，如果您想更改面部颜色，直接分配_facecolor3d就可以了。请注意，您必须为它分配一个rgba_array，例如

scat3D._facecolor3d[0] = [1., 0., 0., 1.]

或者相反，如果您要使用预设（例如“ r”，“。5”等），则可以使用这种方式

scat3D.set_color(...)
scat3D._facecolor3d = scat3D.get_facecolor()
scat3D._edgecolor3d = scat3D.get_edgecolor()

我已经在python 2.7和3.6上进行了测试，没有出现问题。