提取出python streamplot 函数里面的流线,我需要提取出画出图中的某一部分 的流线
可以试试
https://www.jb51.net/article/230885.htm
streamplot函数有返回值,下面的代码以obj为其命名:
import numpy as np
import matplotlib.pyplot as plt
w = 3
Y, X = np.mgrid[-w:w:100j, -w:w:100j]
U = -1 - X**2 + Y
V = 1 + X - Y**2
obj = plt.streamplot(X, Y, U, V, density=[0.5, 1])
plt.show()
将会显示如下结果:
关闭绘图窗口,继续以下代码,就会看到题主想要的内容了:
obj.lines.convert_xunits(X) # 流线的x坐标,二维数组
obj.lines.convert_yunits(Y) # 流线的y坐标,二维数组
obj.arrows.convert_xunits(X) # 箭头的x坐标,二维数组
obj.arrows.convert_yunits(Y) # 箭头的y坐标,二维数组
补充:
1、我想的是将通过U,V画出的流线的其中一部分给提取出来
2、最好能给出一个实例,是如何提取的