gpt4 book ai didi

python - 将交错的 NumPy 整数数组转换为 complex64 的最快方法是什么?

转载 作者:太空狗 更新时间:2023-10-29 20:41:13 25 4
gpt4 key购买 nike

我有一个输入数据流,其中包含交错的实数和虚数整数。将这些转换为 complex64 值是我程序中最慢的操作。这是我目前的做法:

import numpy as np

a = np.zeros(1000000, dtype=np.int16)
b = np.complex64(a[::2]) + np.complex64(1j) * np.complex64(a[1::2])

如果不进行 C 扩展或使用 cython 之类的东西,我能做得更好吗?如果我不能做得更好,使用其中一种技术的最简单方法是什么?

最佳答案

[~]
|1> import numpy as np

[~]
|2> a = np.zeros(1000000, dtype=np.int16)

[~]
|3> b = a.astype(np.float32).view(np.complex64)

[~]
|4> b.shape
(500000,)

[~]
|5> b.dtype
dtype('complex64')

关于python - 将交错的 NumPy 整数数组转换为 complex64 的最快方法是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/5658047/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com