3次元散布図の横にカラーバーを追加するやり方

Question

問題点

3次元散布図の横にカラーバーを追加したいのですが、サイトに載っているやり方を試しても上手くいきません。。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.colors as colors
#############################
n=2
#x,y,z座標
x_list=[i for i in range(2*n+1)]
y_list=[i for i in range(2*n+1)]
z_list=[i for i in range(2*n+1)]
#3次元データ
p_map = np.zeros((2*n+1)*(2*n+1)*(2*n+1)).reshape(2*n+1,2*n+1,2*n+1)
#適当に代入
p_map[0,0,0]=1
p_map[1,3,4]=3
p_map[3,1,0]=8
p_map[2,2,2]=4
#3次元プロット
fig = plt.figure()
ax = fig.add_subplot(111,projection="3d",aspect='equal')
mask = p_map>0
X,Y,Z=np.meshgrid(x_list,y_list,z_list)
ax.set_xlabel("x",labelpad=10,fontsize=24)
ax.set_ylabel("y",labelpad=10,fontsize=24)
ax.set_zlabel("z",labelpad=10,fontsize=24)
sc = ax.scatter(X[mask].ravel(),Y[mask].ravel(),Z[mask].ravel(), p_map[mask].ravel(), cmap="jet")
fig.colorbar(sc)
plt.show()

結果

FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison
  if zdir == 'x':
FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison
  elif zdir == 'y':
Traceback (most recent call last):
    fig.colorbar(sc)
    cb = cbar.colorbar_factory(cax, mappable, **kw)
    cb = Colorbar(cax, mappable, **kwargs)
    mappable.autoscale_None()
    raise TypeError('You must first set_array for mappable')
TypeError: You must first set_array for mappable

すいません、これはどういうことなのでしょうか。。。
ご指導の程宜しくお願い致します。

Answer 1

単に 引数cが設定されてないからではないでしょうか

sc = ax.scatter(X[mask].ravel(),Y[mask].ravel(),Z[mask].ravel(), p_map[mask].ravel(), cmap="jet")

を

sc = ax.scatter(X[mask].ravel(),Y[mask].ravel(),Z[mask].ravel(), c=p_map[mask].ravel(), cmap="jet")

としてみてください

Comments

[Fallout_18]

できました！