import pywt import pandas as pd import numpy as np import matplotlib.pyplot as plt # 设置中文字体为黑体 plt.rcParams['font.sans-serif'] = ['SimHei'] # 读取附件1中的电磁辐射(EMR)数据 emr_data = pd.read_excel('C:/Users\黄某\Downloads\Compressed\e629fcb1-a1a2-4205-916e-c721347f4aa7/2024-51MCM-Problems/2024-51MCM-Problem C/附件1 (Attachment 1).xlsx', sheet_name='EMR') # 提取时间列并排序 emr_data.sort_values(by='时间 (time)', inplace=True) time = pd.to_datetime(emr_data['时间 (time)']) # 提取电磁辐射信号列 emr_signal = emr_data['电磁辐射 (EMR)'].values # 进行小波变换 wavelet = 'db4' # 选择小波函数 coefficients = pywt.wavedec(emr_signal, wavelet) # 绘制小波系数能量谱密度图 plt.figure(figsize=(10, 5)) for i in range(len(coefficients)): plt.plot(np.abs(coefficients[i]), label=f'层级 {i+1}') plt.title('小波系数能量谱密度图') plt.xlabel('系数索引') plt.ylabel('能量') plt.legend() plt.grid(True) # 保存图片为PNG格式
2024五一杯
最新推荐文章于 2024-09-10 13:01:50 发布