配库:
pip install EMD-signal
主要代码:
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()