2024五一杯

最新推荐文章于 2024-09-10 13:01:50 发布

明天知晓

最新推荐文章于 2024-09-10 13:01:50 发布

阅读量377

点赞数 3

文章标签： python

本文链接：https://blog.csdn.net/qq_73376811/article/details/138370805

版权

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格式