[英]How to overlay contour plot on 3-D surface plot with matplotlib or plotly?
我有一個 3-D 表面 plot 顯示 x 和 y 坐標和深度。 我還有一個二維輪廓 plot 與 x 和 y 坐標和不同位置的填充輪廓。 如果我知道輪廓 plot 中坐標處的深度,有沒有辦法可以在 3-D 表面 plot 上顯示輪廓?
我使用plotly
創建了一個 3-D 表面 plot,代碼如下:
import plotly.graph_objects as go
import oceansdb
import numpy as np
import matplotlib.pyplot as plt
Xa = np.linspace(29.005,29.405,200)
Ya = np.linspace(-93.6683,-93.2683,200)
db = oceansdb.ETOPO()
dcont = db['topography'].extract(lat=Xa, lon=Ya)
depth = dcont['height']
fig = go.Figure(data=[go.Surface(z=depth, x=Xa, y=Ya)])
fig.show()
說我的輪廓 plot 可以使用以下代碼創建:
X = np.array([29.1,29.15,29.2,29.25])
Y = np.array([-93.5,-93.45,-93.4,-93.35])
r = np.array([0,0,0,2,3,0,0,6,7,8,9,1,9,0,0,0])
plt.figure()
plt.contourf(X,Y,r.reshape(len(X),len(Y)))
plt.show()
假設可以使用oceansdb
模塊確定每個位置的深度,我可以在表面plot上以正確的深度覆蓋輪廓plot嗎?
使用 matplotlib 的簡短回答是“是”,但有兩個但你有臉:
您需要的關鍵方法是Axes3D.contour
或Axes3D.contourf
。 以下是您的示例數據的實際操作:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D # this enables 3d axes
X = np.array([29.1,29.15,29.2,29.25])
Y = np.array([-93.5,-93.45,-93.4,-93.35])
r = np.array([0,0,0,2,3,0,0,6,7,8,9,1,9,0,0,0]).reshape(X.size, Y.size)
# plot your 2d contourf for reference
fig,ax = plt.subplots()
ax.contourf(X, Y, r)
# plot in 3d using contourf
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.contourf(X, Y, r)
# plot in 3d using contour
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.contour(X, Y, r)
plt.show()
正如您所看到的,后兩者之間的區別在於, contourf
還為每個級別繪制水平面(就像露台一樣),而contour
只繪制水平線本身。
由於使用相同軸的重復繪圖會累積繪圖,因此沒有什么能阻止您在原始表面上疊加這些 3d 等高線圖。 但是,根據我之前的警告,您必須注意輪廓是否在表面上正確渲染(在所有視角下),即使是這樣,結果對於傳達信息也可能不是那么透明。 我個人傾向於發現contour
比contourf
plot 上的輪廓更容易理解,但我懷疑如果我們將這些放在全表面圖的頂部,后者會更好。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.