主页 / 问题 / 611738
Accepted
Алексей Воронов
Asked:2020-01-07 04:43:37 +0000 UTC2020-01-07 04:43:37 +0000 UTC

如何获取一个数组，其中包含另一个数组的复数元素的绝对值？

  • 772

I比方说，有一个由 1000 个元素组成的数组。每个元素都是一个复数。需要得到一个数组J，其元素将是数组中数字的模数的值I。

计算数组值的最佳方法是什么J？循环
计算数组中每个数的模数？I还是可以通过某些功能来做到这一点？

python

3 个回答

  1. Best Answer

    使用 NumPy 模块的解决方案：

    import numpy as np

In [30]: a = np.asarray([complex(1, 1), complex(-1, 2), complex(3, -6)])

In [31]: np.abs(a)
Out[31]: array([ 1.41421356,  2.23606798,  6.70820393])

    性能比较：

    对于 1000 个元素：

    In [17]: a = np.random.randint(-10, 10, 1000) + 1j * np.random.randint(-10, 10, 1000)

In [18]: lst = a.tolist()

In [19]: a.shape
Out[19]: (1000,)

In [20]: len(lst)
Out[20]: 1000

In [21]: %timeit [abs(c) for c in lst]
1000 loops, best of 3: 244 µs per loop

In [22]: %timeit np.abs(a)
10000 loops, best of 3: 28 µs per loop

    对于 1.000.000 个元素：

    In [23]: a = np.random.randint(-10, 10, 10**6) + 1j * np.random.randint(-10, 10, 10**6)

In [24]: lst = a.tolist()

In [25]: a.shape
Out[25]: (1000000,)

In [26]: len(lst)
Out[26]: 1000000

In [27]: %timeit [abs(c) for c in lst]
1 loop, best of 3: 402 ms per loop

In [28]: %timeit np.abs(a)
10 loops, best of 3: 24 ms per loop
    • 4

  2. 您需要的是列表理解。

    它是这样工作的：

    your_list = [complex(1, 1), complex(-1, 2), complex(3, -6)]

abs_list = [abs(c) for c in your_list]

print(abs_list)

    这里第二行完全等同于一个循环：

    abs_list = []
for c in your_list:
    abs_list.append(abs(c))

    在现代 python 中，对于类似的任务，强烈建议使用列表理解，而不是评论中建议的地图。

    • 2

  3. 对于numpy数组：

    >>> import numpy as np
>>> a = np.array([1+1j, -1+2j, 3-6j])
>>> abs(a)
array([ 1.41421356,  2.23606798,  6.70820393])

    对于 Python 列表：

    >>> L = [1+1j, -1+2j, 3-6j]
>>> [abs(c) for c in L]
[1.4142135623730951, 2.23606797749979, 6.708203932499369]
>>> print(*map(abs, L))
1.4142135623730951 2.23606797749979 6.708203932499369

    请参阅*（星号）和 ** 双星号在 Python 中的含义是什么？

    • 2

相关问题