要利用Matplotlib繪制向量場或流場圖,可以使用quiver()
函數。下面是一個簡單的例子:
import numpy as np
import matplotlib.pyplot as plt
# 創建一個二維網格
x = np.linspace(-2, 2, 10)
y = np.linspace(-2, 2, 10)
X, Y = np.meshgrid(x, y)
# 創建一個表示向量場的數據
U = -1 - X**2 + Y
V = 1 + X - Y**2
plt.figure()
plt.quiver(X, Y, U, V)
plt.show()
這段代碼首先創建了一個二維網格,然后根據一個簡單的向量場公式計算出每個位置的向量值,并使用quiver()
函數將向量場繪制出來。可以根據具體的需求調整網格大小、向量場公式和繪圖參數來繪制不同的向量場圖。