Python 使用matplotlib使用剪切单元打印二维阵列

我想在matplotlib中使用pcolor进行打印，类似于链接中的JPEG。我处理的是切割单元，它是2D中的网格单元，被一些障碍物或障碍物一分为二。我需要在每个分裂单元中表示水的状态（高度，速度），所以它们应该可以是不同的颜色。我可以绘制未分割的常规单元格，但我不确定如何绘制切割的单元格

---

您可以先绘制一个普通的彩色网格，然后在其顶部绘制第二个彩色网格，并用多边形剪裁：

将numpy导入为np
将matplotlib.pyplot作为plt导入
从matplotlib.patches导入多边形
图，ax=plt.子批次（）
#第一个网格
a=np.arange（4*5）。重塑（4,5）
ax.pcolormesh（a，cmap='PuBu'，edgecolor='k'，lw=2）
#用于剪裁的多边形
多边形=多边形（xy=np.数组（[[2,4]，[2,0]，[5,4]，[2,4]）、面色='none'，边色='orange'，线宽=2）
ax.添加补片（多边形）
#第二网
b=np.rand.rand（4,5）
网格=ax.pcolormesh（b，cmap='spring'，edgecolor='k'，lw=2）
网格。设置剪辑路径（多边形）
ax.set_xticks（[]）
ax.set_-yticks（[]）
ax.set_方面（“相等”）
plt.show（）

---

谢谢，真的！我只需要弄清楚如何制作这两个阵列