亲宝软件园·资讯

展开

python报错解决方法

shayuxing 人气:0

目前,在使用python处理一个nc文件绘制一个风场图时,出现了以下报错

虽然图片画出来了,但是很丑而且没有理想的填充颜色!

但是不知道为啥,但是参考画图过程,分析这个其中的Z应该指的绘制等高线中的这个函数:matplotlib.pyplot contourf  中使用到的Z!

而这个函数的用法为

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

在这里提出,matplotlib.pyplot contourf 是用来绘制三维等高线图的,不同点是contour()是绘制轮廓线。而contourf()则会填充轮廓。除非有其他说明,否则两个版本的函数是相同的。

参数
X,Y 数组,可选

是在Z中的坐标值。

当X,Y,Z都是二维数组时,它们的形状必须相同,若是一维数组,则X的长度为Z的列数,Y的长度为Z的行数,一般来说创建numpy.meshgrid()

Z 类似矩阵 是绘制轮廓的高度值
levels int或类似数组,可选 用来确定轮廓线\区域的数量和位置
aalpha float,可选 alpha混合值,介于0(透明)和1(不透明)之间。
cmap str或colormap,可选 Colormap用于将数据值(浮点数)从间隔转换为相应Colormap表示的RGBA颜色。用于将数据缩放到间隔中看 。

很显然,在这个函数用法中,如果要画出等高线,核心函数就是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,才发现果然问题出现在这里:

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

这段代码在完成三维数组的绘制时完全没有问题,但是不巧的是,我这次绘制的是四维数组,而其中的ws[i,::]则未取定其中的两维,使得该Z=ws[i,::],理论上来说属于一个三维的数组,所以才会报错。

而我们只需要把这个四维数组取定其中的两维,使得该Z属于一个二维数组即可,已知我的数据中第二维为我恰好需要将之取定为500hpa,所以将这句代码改为:

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

 霍!这不就画出来了嘛!

 初次见面,请多关照!希望能解决你的一点小烦恼哦!

加载全部内容

相关教程
猜你喜欢
用户评论