scipy Numpy：如何将一个nx8数组变成一个nx4数组，其中奇数列是真实的，偶数列是虚数？

cl25kdpy 于 2023-01-13 发布在其他

关注(0)|答案(1)|浏览(180)

我尝试了Numpy: Creating a complex array from 2 real ones?中的解决方案，但似乎无法让它们适用于超过2列的数组。
我的代码可以工作，但效率很低。我无法想象如果列数高得多。

reTrc1_S22_thrutest = thrutest[:,0]
imTrc1_S22_thrutest = 1j*thrutest[:,1]

Trc1_S22_thrutest = reTrc1_S22_thrutest + imTrc1_S22_thrutest

1条答案

要从numpy数组中获取偶数列或奇数列，可以使用slicing syntax。

reTrc1_S22_thrutest = thrutest[:,0::2]
imTrc1_S22_thrutest = 1j*thrutest[:,1::2]

Trc1_S22_thrutest = reTrc1_S22_thrutest + imTrc1_S22_thrutest