【python】将两个实数矩阵合并为一个复数矩阵

来源:互联网 发布:公众号排版 知乎 编辑:程序博客网 时间:2024/05/16 08:47

问题描述:有时需要把两个实数矩阵,一个作为实部,一个作为虚部,合并为一个复数矩阵,该如何操作?
解决办法
假如是在第二个维度上进行合并(real: Data[:, 0, :, :] imag: Data[:, 1, :, :]),有两种方法
一、

result = Data[:, 0, :, :] + 1j*Data[:, 1, :, :]

二、

result = 1j*Data[:, 1, :, :]result += Data[:, 0, :, :]

第二种方法更节省内存

参考链接:https://stackoverflow.com/questions/2598734/numpy-creating-a-complex-array-from-2-real-ones

阅读全文
0 0