小编sha*_*ad0的帖子

如何在 matplotlib 中在等值线图上绘制矢量场?

我已经成功绘制了向量场和等高线图,并希望将它们显示在彼此之上,我环顾四周,但不太明白图形和子图是如何工作的。这是我的代码:

from matplotlib.pyplot import cm
import numpy as np
import matplotlib.pyplot as plt

# Vector Field
Y, X = np.mgrid[-2:2:20j, -2:2:20j]
U =(1 - 2*(X**2))*np.exp(-((X**2)+(Y**2)))
V = -2*X*Y*np.exp(-((X**2)+(Y**2)))
speed = np.sqrt(U**2 + V**2)
UN = U/speed
VN = V/speed
plt.quiver(X, Y, UN, VN, 
           color='Teal', 
           headlength=7)

plt.show()


# Countour Plot
X, Y = np.mgrid[-2:2:100j, -2:2:100j]
Z = X*np.exp(-(X**2 + Y**2))
cp = plt.contourf(X, Y, Z)
plt.colorbar(cp)

plt.show()
Run Code Online (Sandbox Code Playgroud)

python plot matplotlib

4
推荐指数
1
解决办法
1万
查看次数

标签 统计

matplotlib ×1

plot ×1

python ×1