EMD自适应分解Python代码

EMD（Empirical Mode Decomposition）是一种数据分解方法，它将信号分解为一组本地特征模式（Intrinsic Mode Functions，简称IMFs）。下面是一个基于Python的EMD自适应分解的代码示例：

import numpy as np
import scipy.signal as signal

def emd(signal):
    imfs = []  # 存储分解后的IMFs

    while True:
        # 判断是否满足终止条件
        if np.sum(np.abs(signal)) < 1e-10:
            break

        # 计算当前信号的极值点
        maxima = signal.argmax()
        minima = signal.argmin()

        # 提取极值点之间的局部极值包络
        envelope = np.zeros_like(signal)
        envelope[maxima], envelope[minima] = signal[maxima], signal[minima]
        spline = signal[maxima:minima+1]
        interp = np.interp(np.arange(maxima, minima+1), [maxima, minima], spline)
        envelope[maxima:minima+1] = interp

        # 计算当前信号与局部极值包络的差值
        imf = signal - envelope

        # 将提取的IMF存入结果列表中
        imfs.append(imf)

        # 更新当前信号为差值
        signal = imf

    return imfs

# 示例使用
# 生成一个示例信号
t = np.linspace(0, 1, 1000)
signal = np.sin(2 * np.pi * 5 * t) + np.sin(2 * np.pi * 10 * t)

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

# 绘制分解后的IMFs
import matplotlib.pyplot as plt
plt.figure(figsize=(12, 8))
for i, imf in enumerate(imfs):
    plt.subplot(len(imfs), 1, i+1)
    plt.plot(t, imf)
    plt.xlabel('Time')
    plt.ylabel('Amplitude')
plt.tight_layout()
plt.show()

以上代码是一个简单的EMD自适应分解的实现，通过调用`emd(signal)`函数，可以将输入信号分解为一组IMFs。然后，可以对这些IMFs进行进一步的分析或处理。在示例中，我们生成了一个包含两个正弦波的信号，并将其进行了EMD分解，最终绘制了分解后的IMFs。你可以根据自己的需求修改代码。