解压一大堆元组并将其添加到新维度中

Question

我有一个形状为(M,N)的numpy ndarrayx，其中{}中的每个元素都是一个包含复数实部和虚部的2元素元组。在

我想将元组解压到另一个维度中，这样我的x现在的形状是(M,N,2)

我的第一个想法是将实部和虚部分离成两个形状为(M,N)的numpy数组，并将它们连接到第二个轴上，这样我就可以得到所需的输出。为此，我尝试只访问实部和虚部，如下所示

x[:,:][0]
x[:,:][1]

但是，它返回整个数组，而不是只返回实/虚部分。有什么关于我该怎么做的建议吗？在

下面是一个示例数组

^{pr2}$

Answer 1

您可以方便地将数组设置为structured array，这意味着您可以使用每个字段的字符串名访问实部和虚部：

arr['real']
arr['imag']

例如：

^{pr2}$