在python中获取流线的坐标

import numpy as np
import matplotlib.pyplot as plt

# 创建网格
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)

# 定义流线的速度场函数
def velocity_field(x, y):
    u = y
    v = -x
    return u, v

# 计算流线的坐标
U, V = velocity_field(X, Y)

# 绘制流线图
plt.streamplot(X, Y, U, V)
plt.show()

import numpy as np
import plotly.graph_objects as go

# 创建网格
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)

# 定义流线的速度场函数
def velocity_field(x, y):
    u = y
    v = -x
    return u, v

# 计算流线的坐标
U, V = velocity_field(X, Y)

# 创建流线图
fig = go.Figure(data=go.Streamline(x=X.flatten(), y=Y.flatten(), u=U.flatten(), v=V.flatten()))

# 显示图形
fig.show()