擅长:python、mysql、java
<p>您希望脚本返回数组或列表的平方根,但这是不可能的,除非您有一个元素数组或列表!在</p>
<p>例如:</p>
<pre><code>import numpy as np
import math
a = np.array([2, 4, 6])
s = math.sqrt(a)
</code></pre>
<p>这行不通。您将得到相同的错误:</p>
^{pr2}$
<p>必须为数组的每个元素调用<code>math.sqrt</code>函数。。。例如:</p>
<pre><code>for i in a:
s = math.sqrt(i)
print s
</code></pre>
<p>现在你将得到你想要的:</p>
<pre><code>1.41421356237
2.0
2.44948974278
</code></pre>
<hr/>
<p>如果对函数<code>uv2sd</code>做了一些小的更改,代码应该可以工作:</p>
<pre><code>def uv2sd(u,v):
s = []
d = []
for i in range(len(u)):
angulo = math.atan2(u[i],v[i])
if angulo < 0:
angulo = 360 + angulo
d.append(angulo)
s.append(math.sqrt((u[i]*u[i])+(v[i]*v[i])))
return s, d
</code></pre>