如何繪製熱圖的顏色在3D中Matplotlib

Question

我使用Matplotlib 3D繪製我的數據集的3個維度象下面這樣：

但現在我也希望顯示一個第四維（這是與標量值0至20）作爲熱圖。所以基本上，我希望每個點都根據這個第四維的值來選取它的顏色。

Matplotlib中是否存在這樣的事情？如何將[0-20]之間的一串數字轉換爲熱圖顏色？

我把代碼從這裏：http://matplotlib.org/mpl_examples/mplot3d/scatter3d_demo.py

Answer 1

是的，是這樣的：

更新這是一個帶彩條的一個版本。

import numpy as np 
from pylab import * 
from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 

def randrange(n, vmin, vmax): 
    return (vmax-vmin)*np.random.rand(n) + vmin 

fig = plt.figure(figsize=(8,6)) 

ax = fig.add_subplot(111,projection='3d') 
n = 100 

xs = randrange(n, 23, 32) 
ys = randrange(n, 0, 100) 
zs = randrange(n, 0, 100) 

colmap = cm.ScalarMappable(cmap=cm.hsv) 
colmap.set_array(zs) 

yg = ax.scatter(xs, ys, zs, c=cm.hsv(zs/max(zs)), marker='o') 
cb = fig.colorbar(colmap) 

ax.set_xlabel('X Label') 
ax.set_ylabel('Y Label') 
ax.set_zlabel('Z Label') 


plt.show() 

的樣子：

更新下面是一些4維屬性着色數據點的一個明顯的例子。

import numpy as np 
from pylab import * 
from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 

def randrange(n, vmin, vmax): 
    return (vmax-vmin)*np.random.rand(n) + vmin 

fig = plt.figure(figsize=(8,6)) 

ax = fig.add_subplot(111,projection='3d') 
n = 100 

xs = randrange(n, 0, 100) 
ys = randrange(n, 0, 100) 
zs = randrange(n, 0, 100) 
the_fourth_dimension = randrange(n,0,100) 

colors = cm.hsv(the_fourth_dimension/max(the_fourth_dimension)) 

colmap = cm.ScalarMappable(cmap=cm.hsv) 
colmap.set_array(the_fourth_dimension) 

yg = ax.scatter(xs, ys, zs, c=colors, marker='o') 
cb = fig.colorbar(colmap) 

ax.set_xlabel('X Label') 
ax.set_ylabel('Y Label') 
ax.set_zlabel('Z Label') 


plt.show()