Негладкий 3D-график поверхности с использованием PyPlot

Question

Я пытаюсь нарисовать график, подобный этому

Я пришел с крупным планом, но он слишком плавный

# Scene setup
fig = plt.figure(figsize=(12,10))
ax = fig.gca(projection='3d')
ax.view_init(40, 130)

xx, yy = np.meshgrid(np.linspace(-2, 2, 10), np.linspace(-2, 2, 10))
z = phase(activation(cm_num))

# Plot the surface.
surf = ax.plot_surface(xx, yy, z, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

Как мне получить эту ступенчатую форму, или, может быть, мне следует использовать полярные координаты?

Дополнительные примечания: z - это массив пуповинных символов 10x10

array([[3.77, 3.77, 3.77, 3.77, 4.4 , 4.4 , 5.03, 5.03, 5.03, 5.03],
       [3.77, 3.77, 3.77, 3.77, 4.4 , 4.4 , 5.03, 5.03, 5.03, 5.03],
       [3.14, 3.14, 3.77, 3.77, 4.4 , 4.4 , 5.03, 5.03, 5.65, 5.65],
       [3.14, 3.14, 3.14, 3.77, 3.77, 5.03, 5.03, 5.65, 5.65, 5.65],
       [3.14, 3.14, 3.14, 3.14, 3.77, 5.03, 5.65, 5.65, 5.65, 5.65],
       [2.51, 2.51, 2.51, 2.51, 1.88, 0.63, 0.  , 0.  , 0.  , 0.  ],
       [2.51, 2.51, 2.51, 1.88, 1.88, 0.63, 0.63, 0.  , 0.  , 0.  ],
       [2.51, 2.51, 1.88, 1.88, 1.26, 1.26, 0.63, 0.63, 0.  , 0.  ],
       [1.88, 1.88, 1.88, 1.88, 1.26, 1.26, 0.63, 0.63, 0.63, 0.63],
       [1.88, 1.88, 1.88, 1.88, 1.26, 1.26, 0.63, 0.63, 0.63, 0.63]])