python - 将复杂的 NumPy 数组转换为 (n, 2)-实部和虚部数组

Question

我有一个复数值 NumPy 数组，我想将其转换为实部和虚部分开的连续 NumPy 数组。

这个

import numpy

u = numpy.array([
    1.0 + 2.0j,
    2.0 + 4.0j,
    3.0 + 6.0j,
    4.0 + 8.0j
    ])

u2 = numpy.ascontiguousarray(numpy.vstack((u.real, u.imag)).T)

可以解决问题，但是转置、vstacking、和 转换为连续数组可能需要一两步。

是否有一个原生的 NumPy 函数可以帮我做这件事？

Answer 1

您可以使用 column_stack并将两个一维数组堆叠为列以形成单个二维数组。

In [9]: np.column_stack((u.real,u.imag))
Out[9]: 
array([[ 1.,  2.],
       [ 2.,  4.],
       [ 3.,  6.],
       [ 4.,  8.]])