我正在用Python2.7创建一个简单的向量场,然后绘制它
但是Jupyter抱怨除法是0(“RuntimeWarning:divide中遇到的除法是0”),我找不到它
import numpy as np
def field_gen(x0, y0, x, y, q_cons = 1):
dx = x0-x
dy = y0-y
dist = np.sqrt(np.square(dx)+np.square(dy))
kmod = np.where( dist>0.00001, q_cons / dist, 0 )
kdir = np.where( kmod != 0, (np.arctan2(-dy,-dx) * 180 / np.pi), 0)
res_X = np.where( kmod !=0, kmod * (np.cos(kdir)) , 0 )
res_Y = np.where( kmod !=0, kmod * (np.sin(kdir)) , 0 )
return (res_X, res_Y)
n = 10
X, Y = np.mgrid[0:n, 0:n]
x0=2
y0=2
(u,v)= field_gen(x0, y0, X, Y)
#print(u) #debug
#print
#print(v)
plt.figure()
plt.quiver(X, Y, u, v, units='width')
有什么提示吗
别以为
np.where
做了这里所有的工作。在运行调用np.where
之前,Python仍然会首先计算所有输入参数因此,在命令
kmod = np.where( dist>0.00001, q_cons / dist, 0 )
中,Python将计算dist>0.00001
(ok)和q_cons / dist
(bad!)在运行np.where
之前试试np.divide。我想你想要这样的东西:
相关问题 更多 >
编程相关推荐