3D图+热图中的图4变量

Question

我在一个数据帧中有4列，我想看看它们之间是否存在相关性。我认为它可以给我一些洞察力，通过将它们绘制在3D图中，然后添加第四个维度作为热图，但我不知道如何添加链接到数据帧中的一列的热图。

这是我到目前为止所得到的：

from mpl_toolkits.mplot3d import Axes3D  
fig = plt.figure(figsize=(12,12));
ax = fig.add_subplot(111, projection='3d');
ax.scatter(Series1,Series2,Series3); 

这将返回给我以下内容：

(https://i.stack.imgur.com/qRGMv.png)

所以，上面有Series1，Series2和Series3的观点...但是有没有一种方法可以添加一个热图或任何东西来区别于Series4呢？

Answer 1

我发现有一种方法可以通过添加c参数来添加色彩映射表，如下所示：

withcolormap = ax.scatter(Series1, Series2, Series3, c = Series4, cmap='gist_heat');

然后把它的柱状图放在一边：

fig.colorbar(withcolormap, shrink=0.75);

由于它给了我一些见解，我想知道如何绘制这个图的旋转视图的动画，我通过执行以下代码进行了尝试：

from matplotlib.animation import FuncAnimation, PillowWriter

def rotation(i):
  ax = fig.add_subplot(111, projection='3d')
  ax.view_init(elev=30, azim=i);

ani = FuncAnimation(withcolormap, rotation, frames=range(0,360,10))

然后返回一个错误：

AttributeError: 'Path3DCollection' object has no attribute 'canvas'