如何利用Matplotlib繪制向量場(chǎng)或流場(chǎng)圖

小億
125
2024-05-20 12:55:31

要利用Matplotlib繪制向量場(chǎng)或流場(chǎng)圖，可以使用quiver()函數(shù)。下面是一個(gè)簡(jiǎn)單的例子：

import numpy as np
import matplotlib.pyplot as plt

# 創(chuàng)建一個(gè)二維網(wǎng)格
x = np.linspace(-2, 2, 10)
y = np.linspace(-2, 2, 10)
X, Y = np.meshgrid(x, y)

# 創(chuàng)建一個(gè)表示向量場(chǎng)的數(shù)據(jù)
U = -1 - X**2 + Y
V = 1 + X - Y**2

plt.figure()
plt.quiver(X, Y, U, V)
plt.show()

這段代碼首先創(chuàng)建了一個(gè)二維網(wǎng)格，然后根據(jù)一個(gè)簡(jiǎn)單的向量場(chǎng)公式計(jì)算出每個(gè)位置的向量值，并使用quiver()函數(shù)將向量場(chǎng)繪制出來(lái)?？梢愿鶕?jù)具體的需求調(diào)整網(wǎng)格大小、向量場(chǎng)公式和繪圖參數(shù)來(lái)繪制不同的向量場(chǎng)圖。

0 踩