plt.subplot(4, 4, i + 1) plt.title(f) plt.pcolor(W[:, :, i].T, cmap='rainbow')设置颜色映射范围

可以使用`vmin`和`vmax`参数来设置颜色映射范围，例如：

plt.subplot(4, 4, i + 1)
plt.title(f)
plt.pcolor(W[:, :, i].T, cmap='rainbow', vmin=0, vmax=1)

这里将最小值设置为0，最大值设置为1，使得颜色映射范围在0到1之间。你可以根据具体情况调整这两个参数。