矩阵分解emd

import numpy as np
import matplotlib.pyplot as plt
from PyEMD import EMD

# 生成示例数据
t = np.linspace(0, 1, 1000)
signal = np.sin(2 * np.pi * 5 * t) + np.sin(2 * np.pi * 10 * t)

# 创建EMD对象
emd = EMD()

# 执行EMD分解
imfs = emd(signal)

# 绘制原始信号和IMFs
plt.figure(figsize=(10, 6))

plt.subplot(len(imfs) + 1, 1, 1)
plt.plot(t, signal, 'r')
plt.title("Original Signal")

for i, imf in enumerate(imfs):
    plt.subplot(len(imfs) + 1, 1, i + 2)
    plt.plot(t, imf, 'g')
    plt.title(f"IMF {i+1}")

plt.tight_layout()
plt.show()