我需要绘制一个分段流图,其中我的向量场由
def f(x):
if x<= 2:
return 5/((3-2*(-x))**2)-((-18+(-x))*(-6+(-x)))/((-x)**4)
else:
return 5/((3-2*x)**2)-((-18+x)*(-6+x))/(x**4)
u = y
v = f(x)
我怎么能执行这个密谋?
发布于 2020-01-08 11:55:29
使用np.piecewise
定义函数和np.mgrid
import numpy as np
import matplotlib.pyplot as plt
w = 3
y, x = np.mgrid[-w:w:100j, -w:w:100j]
U = y
V = np.piecewise(x, [x <= 2, x > 2], [lambda x: 5/((3-2*(-x))**2)-((-18+(-x))*(-6+(-x)))/((-x)**4), lambda x: 5/((3-2*x)**2)-((-18+x)*(-6+x))/(x**4)])
plt.streamplot(x, y, U, V, density=[0.5, 1])
plt.show()
给出了以下情节:
https://stackoverflow.com/questions/59645075
复制相似问题