python 如何将两个实数矩阵合并为一个复数矩阵

问题描述：

有时需要把两个实数矩阵，一个作为实部，一个作为虚部，合并为一个复数矩阵，该如何操作？

解决办法：

假如是在第二个维度上进行合并(real: data[:, 0, :, :] imag: data[:, 1, :, :])，有两种方法

第一种、

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

第二种、

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

第二种方法更节省内存~

补充：python numpy 分离与合并复数矩阵实部虚部的方法

在进行数字信号处理的过程中，我们往往有对短时傅里叶变换频谱(spectrogram)进行分析的需求。

常见的分析手段对应欧拉公式分为两种，要么使用模与相位的形式，要么使用实部虚部。

本文分享一个简单的将复数光谱图分解为实部与虚部以及将两个部分重新合并为一个复数矩阵的过程，以下为python代码。

import numpy as np
import librosa

# load the original wav
test_wave, _ = librosa.load("../recfile_1_20200617_153719_sound_capture_dshow_5_monooutput1.wav", sr=44100)
# calculate the complex spectrogram stft
spectrogram_test_wav = librosa.stft(test_wave, n_fft=735*2, win_length=735*2, hop_length=735)

# calculate the real part of the spectrogram
real_spectrogram = spectrogram_test_wav.real
# calculate the imaginary part of the spectrogram
imaginary_spectrogram = spectrogram_test_wav.imag

# combine these two parts
reconstruction_spectrogram = real_spectrogram + 1j * imaginary_spectrogram
print(np.array_equal(spectrogram_test_wav, reconstruction_spectrogram))

其中librosa库为常用的音频处理库。

上述代码实现了对wavfile进行短时傅里叶变换，分离出实部虚部并重新合并的过程。

最终的输出为true, 证明了经过这些步骤过后，重构的复数矩阵与初始的光谱图是一致的。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。