如何连接和绘制超过特定阈值的 2D 切片部分？

Question

我想在曲面图中连接二元曲面，以实现 3D 连接轮廓。

到目前为止，我在这个最小示例中绘制了 2D 切片：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d   import Axes3D
from matplotlib.collections import PolyCollection
import matplotlib
matplotlib.use('Qt5Agg')

x, y = np.meshgrid(np.linspace(0,1,20), np.linspace(0,1,20))
z = np.linspace(0, 2, 4)

u = np.zeros((len(z),20,20))

for i in range(len(z)):
    u[i, i+5:i+7, i+5:i+7] = 1

ax = plt.figure().add_subplot(111, projection = '3d')
for i in range(len(z)):
    im = ax.pcolor(x, y, u[i,:,:], alpha=0.1) 
    ax.add_collection3d(im, zs=z[i], zdir='z')    
plt.show()

有人知道如何将这些黄色区域连接到 3D 轮廓并绘制出来吗？ 多谢！ 这是我想要获得的示例：