Какой самый быстрый способ преобразовать массив чередующихся чисел NumPy в complex64?

Question

У меня есть поток входящих данных, который чередовал реальные и мнимые целые числа. Преобразование их в значения 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? Если я не могу добиться большего успеха, каков мой самый простой подход с использованием технологии, подобной одной из них?

Answer 1

[~]
|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')