Это создает цветную панель (хотя, возможно, и не тот, который вам нужен):
Замените эту строку:
ax.scatter(xs, ys, zs, c=cs, marker=m)
с
p = ax.scatter(xs, ys, zs, c=cs, marker=m)
, затем используйте
fig.colorbar(p)
ближе к концу