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