目前网上的办法,直接导入:from PyEMD import EMD, Visualisation 是有问题的
可能会出现 在 ‘init.py | init.py’ 中找不到引用 ‘Visualisation’ 的报错。
原因似乎是现在导入的命令改了,这是一个坑,解决的办法如下:
import numpy as np
import matplotlib.pyplot as plt
from PyEMD.EMD import EMD
from PyEMD.visualisation import Visualisation
# 创建一个示例信号
t = np.linspace(0, 1, 1000)
signal = np.sin(20 * np.pi * t) + np.sin(5 * np.pi * t)
# 创建EMD对象
emd = EMD()
# 对信号进行EMD分解
IMFs = emd(signal)
# 绘制原始信号和分解的IMFs
plt.figure(figsize=(12, 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)
plt.title(f"IMF {i+1}")
plt.tight_layout()
plt.show()