Streamtube類是用于可視化流線的Plotly圖表類型之一。它可以在3D圖表中顯示流線,并用不同顏色表示流體的速度。要使用Streamtube類,首先需要導入plotly.graph_objects模塊,然后創建一個Figure對象并添加Streamtube對象。
下面是一個簡單的示例代碼,演示如何使用Streamtube類創建一個流線圖表:
import plotly.graph_objects as go
import numpy as np
# 創建數據
x, y, z = np.mgrid[-10:10:20j, -10:10:20j, -10:10:20j]
u = -y
v = x
w = np.zeros_like(x)
# 創建Figure對象并添加Streamtube對象
fig = go.Figure(data=go.Streamtube(x=x.flatten(), y=y.flatten(), z=z.flatten(), u=u.flatten(), v=v.flatten(), w=w.flatten(), colorscale='Blues'))
# 設置圖表布局
fig.update_layout(scene=dict(aspectratio=dict(x=1, y=1, z=1), camera_eye=dict(x=1.87, y=0.88, z=-0.64)))
# 顯示圖表
fig.show()
在這個示例中,我們創建了一個三維網格數據,并定義了流線的速度向量。然后我們使用Streamtube對象將流線添加到Figure對象中,并設置相應的顏色映射。最后,我們更新圖表的布局并顯示圖表。
通過調整數據和參數,您可以創建不同形狀和風格的流線圖表。您還可以進一步自定義圖表,以滿足您的需求和偏好。