python報(bào)錯(cuò)TypeError: Input z must be 2D, not 3D的解決方法

更新時(shí)間：2021年12月20日 10:30:00 作者：shayuxing

大家好，本篇文章主要講的是python報(bào)錯(cuò)TypeError: Input z must be 2D, not 3D的解決方法，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話記得收藏一下

目前，在使用python處理一個(gè)nc文件繪制一個(gè)風(fēng)場(chǎng)圖時(shí)，出現(xiàn)了以下報(bào)錯(cuò)

雖然圖片畫出來(lái)了，但是很丑而且沒(méi)有理想的填充顏色！

但是不知道為啥，但是參考畫圖過(guò)程，分析這個(gè)其中的Z應(yīng)該指的繪制等高線中的這個(gè)函數(shù)：matplotlib.pyplot contourf? 中使用到的Z！

而這個(gè)函數(shù)的用法為

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

在這里提出，matplotlib.pyplot contourf?是用來(lái)繪制三維等高線圖的，不同點(diǎn)是contour（）是繪制輪廓線。而contourf()則會(huì)填充輪廓。除非有其他說(shuō)明，否則兩個(gè)版本的函數(shù)是相同的。

參數(shù)
X,Y	數(shù)組，可選	是在Z中的坐標(biāo)值。當(dāng)X,Y,Z都是二維數(shù)組時(shí)，它們的形狀必須相同，若是一維數(shù)組，則X的長(zhǎng)度為Z的列數(shù)，Y的長(zhǎng)度為Z的行數(shù)，一般來(lái)說(shuō)創(chuàng)建numpy.meshgrid()
Z	類似矩陣	是繪制輪廓的高度值
levels	int或類似數(shù)組，可選	用來(lái)確定輪廓線\區(qū)域的數(shù)量和位置
aalpha	float,可選	alpha混合值，介于0（透明）和1（不透明）之間。
cmap	str或colormap，可選	Colormap用于將數(shù)據(jù)值（浮點(diǎn)數(shù)）從間隔轉(zhuǎn)換為相應(yīng)Colormap表示的RGBA顏色。用于將數(shù)據(jù)縮放到間隔中看?。

很顯然，在這個(gè)函數(shù)用法中，如果要畫出等高線，核心函數(shù)就是plt.contourf()，但在這個(gè)函數(shù)中輸入的參數(shù)是x,y對(duì)應(yīng)的網(wǎng)格數(shù)據(jù)以及此網(wǎng)格對(duì)應(yīng)的高度值，因此我們調(diào)用np.meshgrid(x,y)把x,y值轉(zhuǎn)換成網(wǎng)格數(shù)據(jù)：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
  
# 計(jì)算x,y坐標(biāo)對(duì)應(yīng)的高度值
def f(x, y):
 return (1-x/2+x**3+y**5) * np.exp(-x**2-y**2)
 
# 生成x,y的數(shù)據(jù)
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
 
# 把x,y數(shù)據(jù)生成mesh網(wǎng)格狀的數(shù)據(jù)，因?yàn)榈雀呔€的顯示是在網(wǎng)格的基礎(chǔ)上添加上高度值
X, Y = np.meshgrid(x, y)
 
# 填充等高線
plt.contourf(X, Y, f(X, Y))
# 顯示圖表
plt.show()

于是我分析了我的代碼中的contourf中對(duì)應(yīng)的X,Y和Z，才發(fā)現(xiàn)果然問(wèn)題出現(xiàn)在這里：

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

這段代碼在完成三維數(shù)組的繪制時(shí)完全沒(méi)有問(wèn)題，但是不巧的是，我這次繪制的是四維數(shù)組，而其中的ws[i,::]則未取定其中的兩維，使得該Z=ws[i,::]，理論上來(lái)說(shuō)屬于一個(gè)三維的數(shù)組，所以才會(huì)報(bào)錯(cuò)。

而我們只需要把這個(gè)四維數(shù)組取定其中的兩維，使得該Z屬于一個(gè)二維數(shù)組即可，已知我的數(shù)據(jù)中第二維為我恰好需要將之取定為500hpa，所以將這句代碼改為：

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

?霍！這不就畫出來(lái)了嘛！

?初次見面，請(qǐng)多關(guān)照！希望能解決你的一點(diǎn)小煩惱哦！

到此這篇關(guān)于python報(bào)錯(cuò)TypeError: Input z must be 2D, not 3D的解決方法的文章就介紹到這了,更多相關(guān)python報(bào)錯(cuò)解決方法內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: