Numpy在meshgrid中迭代计算

-1

最近，通过一个stackoverflow成员的友好响应，我学会了如何使用for循环技巧迭代遍历每一个函数，使用与numpy不连续的函数在meshgrid上执行操作网格中的元素。然而，我发现它在大多数情况下都能正常工作，但由于某种原因，它似乎失败了：
Emag是我创建的一个长函数。Numpy在meshgrid中迭代计算

import numpy as np 
X, Y = np.mgrid[0.1:0.15:3j, 0.1:0.15:3j] 
Z = np.zeros_like(Y)      
po = np.sqrt(X**2 + Y**2) 
phio = np.arctan2(Y,X) 
print po 
print phio 

for i in range(po.shape[0]):      
    for j in range(po.shape[1]): 
     print po[i,j] 
     print phio[i,j] 
     Z[i,j] = Emag(po[i,j], phio[i,j], 0) 
     print Z[i,j]

下面是响应

%run C:/Users/Nigel/Desktop/Nigel07112014.py 
[[ 0.14142136 0.16007811 0.18027756]   <--- po 
[ 0.16007811 0.1767767 0.19525624] 
[ 0.18027756 0.19525624 0.21213203]] 
[[ 0.78539816 0.89605538 0.98279372]   <--- phio 
[ 0.67474094 0.78539816 0.87605805] 
[ 0.5880026 0.69473828 0.78539816]] 
0.141421356237 <--- po[0,0] 
0.785398163397 <--- phio[0,0] 
732.46186213  <--- Z[0,0] 
0.160078105936 <--- po[0,1] 
0.896055384571 <--- phio[0,1] 
--------------------------------------------------------------------------- 
OverflowError        Traceback (most recent call last)

似乎为meshgrid的第二元件，所述计算机具有用于产生Z.输出麻烦然而，当我执行埃马克对于第二元件PO和phio分别 - 它工作得很好（见下文）：

Emag(0.160078105936, 0.896055384571, 0) 
> 589.541876301

这应该是在Z meshgrid的第二元件。

那么哪里出问题了？如果你处理的虚参数的指数

--------------------------------------------------------------------------- 
OverflowError        Traceback (most recent call last) 
C:\Users\Nigel\AppData\Local\Enthought\Canopy\App\appdata\canopy-1.4.1.1975.win-x86_64\lib\site-packages\IPython\utils\py3compat.pyc in execfile(fname, glob, loc) 
    195    else: 
    196     filename = fname 
--> 197    exec compile(scripttext, filename, 'exec') in glob, loc 
    198  else: 
    199   def execfile(fname, *where): 

C:\Users\Nigel\Desktop\Nigel07112014.py in <module>() 
    220   print po[i,j] 
    221   print phio[i,j] 
--> 222   Z[i,j] = Emag(po[i,j], phio[i,j], 0) 
    223   print Z[i,j] 
    224 

C:\Users\Nigel\Desktop\Nigel07112014.py in Emag(p, phi, z) 
    149 def Emag(p, phi, z): 
    150  if z >=0: 
--> 151   VE1p = (p * (z-z0) * ((MuZ)/(4 * pi * Eps0 * Eps1)) * exp(j * k1 * R0/(R0**3)) * ((3/(R0**2))- (3 * j * k1/R0)-(k1**2))) - (j * ((MuZ)/(4 * pi * Eps0 * Eps1)) * INTEGRATEZI(lambda kp: J1(kp * p) * A1(kp) * kp * k1z(kp) * (exp(1j * k1z(kp) * (z + z0))))) 
    152 
    153   VE1phi = 0 

OverflowError: math range error

来源

2014-11-23 user3065619

作为一个快速调试提示，修改'Emag'来打印它的参数。 – gboffi 2014-11-23 16:17:00

对于它的价值，我尝试了一个模拟Emag的代码，它的运行完美无缺......也许你必须发布完整的追溯 – gboffi 2014-11-23 16:19:30

嗨gboffi。我已经添加了完整的错误信息 - 请记住，Emag确实为单个值工作 - 只是当它们位于meshgrid中时才会工作 – user3065619 2014-11-23 16:37:56

变化exp(j ...在Emag到exp(1j ...：若跌破

完整的错误讯息的要求我会贴更长的错误消息。当它在循环内部执行时j可以收到不好的值从j循环的索引。

对于虚构的论点，我应该是正确的，我也会三重检查您的Emag函数的结果。

来源

2014-11-23 17:03:39 gboffi

嗨。在我写的代码的开头j = 1j（上面没见过） – user3065619 2014-11-23 17:06:16

它也没有解释 - 我想 - 为什么当输入值时分开它工作正常。 – user3065619 2014-11-23 17:12:07

即使您试图设置'j = 1j'，如果您稍后使用名为'j'的变量进行循环，它会一遍又一遍地破坏性地更改该值。您将无法保证以后这是复杂的单位。真的，你需要更加谨慎地使用代码中的复数。 – ely 2014-11-23 18:07:31

Numpy在meshgrid中迭代计算

回答

相关问题