pyhht安裝，直接pip install pyhht即可
# -*- coding: utf-8 -*-
import numpy as np
import pandas as pd
from pyhht.emd import EMD
from pyhht.visualization import plot_imfs
from sql_bind_data import sum_id_value,filter_out
values1=sum_id_value(2)
values1=filter_out(values1)
values1.fillna(0,inplace=True)#將數據中的nan替換，EMD不支持處理nan
data=values1
decomposer = EMD(data[0])
imfs = decomposer.decompose()
#繪製分解圖
plot_imfs(data[0],imfs,data.index)
#保存IMFs
arr = np.vstack((imfs,data[0]))
dataframe = pd.DataFrame(arr.T)
dataframe.to_csv('D:/GitHub/模態分解成分.csv',index=None,columns=None)
