我有一个形状为(M,N)
的numpy ndarrayx
,其中{}中的每个元素都是一个包含复数实部和虚部的2元素元组。在
我想将元组解压到另一个维度中,这样我的x
现在的形状是(M,N,2)
我的第一个想法是将实部和虚部分离成两个形状为(M,N)
的numpy数组,并将它们连接到第二个轴上,这样我就可以得到所需的输出。为此,我尝试只访问实部和虚部,如下所示
x[:,:][0]
x[:,:][1]
但是,它返回整个数组,而不是只返回实/虚部分。有什么关于我该怎么做的建议吗?在
下面是一个示例数组
^{pr2}$
Tags:
你可以尝试使用下面的代码
对于真实的部分和想象的部分
^{pr2}$您可以方便地将数组设置为structured array,这意味着您可以使用每个字段的字符串名访问实部和虚部:
例如:
^{pr2}$相关问题 更多 >
编程相关推荐