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

2024-03-28 08:31:28 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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

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

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

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

下面是一个示例数组

^{pr2}$

Tags: numpy元素示例数组建议复数元组形状
2条回答

你可以尝试使用下面的代码

[x[0] for x in arr]

对于真实的部分和想象的部分

^{pr2}$

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

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

例如:

^{pr2}$

相关问题 更多 >