ax.scatter(p.ravel(), t.ravel(), s=60, c=d.ravel(), cmap=plt.cm.coolwarm,
    norm=norm, clip_on=False)
ax.set_xlabel('$\\Phi$', fontsize='xx-large')
ax.set_ylabel('$\\Theta$', fontsize='xx-large')
ax.set_yticks(np.linspace(0, np.pi, 3))
ax.set_yticklabels(['$0$', '$\\frac{\\pi}{2}$', '$\\pi$'], fontsize='x-large')
ax.set_xticks(np.linspace(0, 2 * np.pi, 5))
ax.set_xticklabels(['$0$', '$\\frac{\\pi}{2}$', '$\\pi$',
    '$\\frac{3\\pi}{4}$', '$2\\pi$'], fontsize='x-large')
ax.set_xlim(0, 2 * np.pi)
ax.set_ylim(0, np.pi)
cb = plt.colorbar(im)
cb.set_label('Voltage', fontsize='x-large')
fig.tight_layout()
Добавление интенсивности к 3D-графику