Python报错TypeError: Input z must be 2D, not 3D的解决方法

问题现象与错误分析

在利用Python处理.nc格式文件并绘制风场图的过程中，程序抛出了如下错误信息。

尽管图像最终生成，但视觉效果不佳，且未能实现预期的颜色填充效果。

经过对绘图流程的复盘，推测该错误源于matplotlib.pyplot模块中contourf函数所接收的Z参数维度不符。

具体而言，该函数的调用语法涉及对网格坐标及高度值的指定。

coutour([X, Y,] Z,[levels], **kwargs)

需要明确的是，matplotlib.pyplot.contourf用于绘制填充等高线图，而contour仅绘制轮廓线。除填充效果外，两者的基本用法一致。

contourf函数原理与网格生成

要实现等高线绘制，核心在于调用plt.contourf()。该函数要求输入x和y对应的网格数据，以及网格上对应的高度值。因此，通常需使用np.meshgrid(x, y)将一维的x和y坐标转换为二维网格数据。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
  
# 计算x,y坐标对应的高度值
def f(x, y):
 return (1-x/2+x**3+y**5) * np.exp(-x**2-y**2)
 
# 生成x,y的数据
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
 
# 把x,y数据生成mesh网格状的数据，因为等高线的显示是在网格的基础上添加上高度值
X, Y = np.meshgrid(x, y)
 
# 填充等高线
plt.contourf(X, Y, f(X, Y))
# 显示图表
plt.show()

错误原因定位与修正方案

基于上述原理，我对代码中传入contourf函数的X、Y及Z参数进行了检查，确认问题出在Z参数的维度上。

 cp=ax.contourf(lon,lat,ws[i,::],zorder=0,transform=ccrs.PlateCarree(),cmap=cmap,levels=levels,extend='both')

原代码在处理三维数组时运行正常，但本次数据为四维数组。由于代码中ws[i,::]未完全指定维度，导致Z变量实际上仍是一个三维数组，从而触发了维度不匹配的错误。

解决方法是将四维数组进一步切片，提取出二维子集作为Z。鉴于数据中第二维代表500hpa气压层，我将相关代码修改为明确指定该维度索引。

cp=ax.contourf(lon,lat,ws[i,0,::],zorder=0,transform=ccrs.PlateCarree(),cmap=cmap,levels=levels,extend='both')

修改后，图像成功渲染，填充效果符合预期。

希望这篇解答能为您解决类似的技术困扰提供协助。

以上就是关于Python中TypeError: Input z must be 2D, not 3D错误的完整解决方案。如需了解更多Python报错处理技巧，欢迎查阅本站历史文章或继续浏览后续内容。感谢您的支持。