如何用streamplot绘制流图

「已注销」

已于 2023-11-19 08:25:28 修改

阅读量438

点赞数

文章标签： python

于 2023-11-19 08:24:50 首次发布

本文链接：https://blog.csdn.net/m0_38127487/article/details/134486834

版权

from turtle import color
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D

plt.style.use('seaborn-darkgrid')

# make a stream function:
X, Y = np.meshgrid(np.linspace(-3, 3, 256), np.linspace(-3, 3, 256))
Z = (1 - X/2 + X**5 + Y**3) * np.exp(-X**2 - Y**2)
# make U and V out of the streamfunction:
V = np.diff(Z[1:, :], axis=1)
U = -np.diff(Z[:, 1:], axis=0)

# plot:
fig, ax = plt.subplots()

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='Accent')

plt.show()

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = V,cmap='Accent')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='Accent_r')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='Paired')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='Paired_r')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='BrBG')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='BrBG_r')

在这里插入图片描述

ax.streamplot(X[1:, 1:], Y[1:, 1:],U, V,color = U,cmap='BuGn')

在这里插入图片描述

参考文献:https://vimsky.com/examples/usage/matplotlib-pyplot-streamplot-in-python.html
https://matplotlib.org/stable/plot_types/arrays/streamplot.html#sphx-glr-plot-types-arrays-streamplot-py