我知道 Python 有 cmath 模块来查找负数的平方根。
我想知道的是，如何对一个包含 100 个负数的数组做同样的事情？
问问题
9451 次
3 回答
7
您想遍历列表的元素并sqrt在其上应用函数。您可以使用内置函数map，它将第一个参数应用于第二个参数的每个元素：
lst = [-1, 3, -8]
results = map(cmath.sqrt, lst)
另一种方法是使用经典的列表推导：
lst = [-1, 3, -8]
results = [cmath.sqrt(x) for x in lst]
执行示例：
>>> lst = [-4, 3, -8, -9]
>>> map(cmath.sqrt, lst)
[2j, (1.7320508075688772+0j), 2.8284271247461903j, 3j]
>>> [cmath.sqrt(x) for x in lst]
[2j, (1.7320508075688772+0j), 2.8284271247461903j, 3j]
如果您使用的是 Python 3，则可能必须应用list()map 的结果（或者您将拥有一个 ietrator 对象）
于 2014-01-21T16:08:34.197 回答
3
import cmath, random
arr = [random.randint(-100, -1) for _ in range(10)]
sqrt_arr = [cmath.sqrt(i) for i in arr]
print(list(zip(arr, sqrt_arr)))
结果：
[(-43, 6.557438524302j), (-80, 8.94427190999916j), (-15, 3.872983346207417j), (-1, 1j), (-60, 7.745966692414834j), (-29, 5.385164807134504j), (-2, 1.4142135623730951j), (-49, 7j), (-25, 5j), (-45, 6.708203932499369j)]
于 2014-01-21T16:07:55.263 回答
3
如果速度是一个问题，您可以使用 numpy：
import numpy as np
a = np.array([-1+0j, -4, -9])
np.sqrt(a)
# or:
a**0.5
结果：
array([ 0.+1.j, 0.+2.j, 0.+3.j])
于 2014-01-21T17:29:22.553 回答