如何在 2D 中投影 3D 球函数？

Question

我有这段代码来绘制 3D 球体函数的图形：

import numpy as np
import matplotlib.pyplot as plt

t = np.linspace(-5.12,5.12, 100)
y = np.linspace(-5.12,5.12, 100)
X, Y = np.meshgrid(t, y)
Z = X ** 2 + Y ** 2
fig = plt.figure()
ax = plt.axes(projection ='3d')
ax.plot_surface(X, Y, Z, cmap ='viridis', edgecolor ='c')
ax.set_title('3D Sphere Function')
plt.show()

现在我想在图中显示一组点，但它不会很明显。我想投影相同的球体，但在平面 2D 表面上，第三维投影为颜色，如下所示：

我怎么能用 python 绘制这个？