要利用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()
函数将向量场绘制出来。可以根据具体的需求调整网格大小、向量场公式和绘图参数来绘制不同的向量场图。