Matplotlib 3d рассеянная карта цвета

Question

Я опубликовал ранее о программе, которую я пытался сделать, чтобы построить четырехмерные данные (координаты и напряжения xyz) на графике 3d-рассеяния с цветовой картой, зависящей от величины напряжений.

Я могу запустить свою программу без ошибок (предполагая, что я не пропускаю никаких пакетов), хотя мои точки данных всегда выглядят голубыми, что указывает на то, что цветовая карта не применяется должным образом. Кто-нибудь может понять, в чем проблема? Я очень признателен, если так.

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

x = array([-4,-4, -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,4,4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, -5, -4.5, -4, -3.5, -3, -2.5, -2, -1.5, -1, -0.5, 0, 0.5, 1, 1.5, 2, 2.5, 3, 3.5, 4, 4.5, 5, 5.5, 6, 6.5, -6.5, -6, -5.5, 
    ]) 
y = array([-0.1, 3.95, 5.8,4.4,0.1,-4,-5.8,-4,1,2.4,3.2, 1.6,-0.8,-2.6,-3.3,-1.4,-0.1,3.95, 5.8,4.4,0.1,-4,-5.8,-4,1,2.4, 3.2, 1.6, -0.8, -2.6, -3.3, -1.4, -0.1, 3.95, 5.8, 4.4, 0.1, -4, -5.8, -4, 1, 2.4, 3.2, 1.6, -0.8, -2.6, -3.3, -1.4, -0.1, 3.95, 5.8, 4.4, 0.1, -4, -5.8, -4, 1, 2.4, 3.2, 1.6, -0.8, -2.6, -3.3, -1.4, -0.1, 3.95, 5.8, 4.4, 0.1, -4, -5.8, -4, 1, 2.4, 3.2, 1.6, -0.8, -2.6, -3.3, -1.4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
    ]) 
z = array([5.85, 4, 0.2,-3.8,-5.85,-4.1,-0.15,4,2.75,1.4,-0.3, -2.6, -2.75,-2.7, 1.2, 3.2,5.85,4, 0.2,-3.8,-5.85,-4.1,-0.15, 4, 2.75, 1.4, -0.3, -2.6, -2.75, -2.7, 1.2, 3.2, 5.85, 4, 0.2, -3.8, -5.85, -4.1, -0.15, 4, 2.75, 1.4, -0.3, -2.6, -2.75, -2.7, 1.2, 3.2, 5.85, 4, 0.2, -3.8, -5.85, -4.1, -0.15, 4, 2.75, 1.4, -0.3, -2.6, -2.75, -2.7, 1.2, 3.2, 5.85, 4, 0.2, -3.8, -5.85, -4.1, -0.15, 4, 2.75, 1.4, -0.3, -2.6, -2.75, -2.7, 1.2, 3.2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
    ]) 
v = array([0.29, 0.32, 0.3, 0.27, 0.3, 0.28, 0.31, 0.28, 0.19, 0.18, 0.19, 0.18, 0.19, 0.19, 0.2, 0.19, 0.12, 0.12, 0.11, 0.1, 0.11, 0.15, 0.14, 0.13, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.32, 0.31, 0.3, 0.31, 0.29, 0.3, 0.32, 0.31, 0.2, 0.21, 0.19, 0.19, 0.19, 0.19, 0.19, 0.2, 0.13, 0.12, 0.14, 0.14, 0.17, 0.17, 0.16, 0.14, 0.18, 0.18, 0.18, 0.19, 0.19, 0.19, 0.19, 0.18, 0.16, 0.18, 0.2, 0.19, 0.18, 0.15, 0.16, 0.18, 0.18, 0.18, 0.18, 0.17, 0.17, 0.17, 0.18, 0.18, 0.13, 0.14, 0.15, 0.16, 0.17, 0.17, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.18, 0.17, 0.17, 0.16, 0.15, 0.14, 0.12, 0.08, 0.09, 0.11, 0.09, 0.1, 0.11, 
    ]) 

c = abs(v) 
fig = plt.figure() 
ax = fig.add_subplot(111, projection='3d') 
cmhot = plt.get_cmap("hot") 
cax = ax.scatter(x, y, z, v, s=50, c = c, cmap = cmhot) 
plt.show() 

Большое спасибо заранее, кто бы ни был любезен, чтобы посмотреть на это для меня.

Answer 1

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

th = np.linspace(0, 2 * pi, 100) 
x = cos(th) 
y = sin(th) 
z = th 
v = cos(th) * sin(th) 
c = abs(v) 
fig = plt.figure() 
ax = fig.add_subplot(111, projection='3d') 
cmhot = plt.get_cmap("hot") 
cax = ax.scatter(x, y, z, v, s=50, c = c, cmap = cmhot) 
plt.show() 

Производит

Смотрите эту Matplotlib 3D scatter color lost after redraw, который сообщает ту же самую вещь, начиная с версии 1.1.0 (и имеет обходным). Это старая ошибка в mpl, которая была исправлена ​​в более поздних версиях (PR here)