Python:从seaborn kdeplot获取FWHM

import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import find_peaks

# 生成一些示例数据
data = np.random.normal(loc=0, scale=1, size=1000)

# 绘制核密度估计图
sns.kdeplot(data, shade=True)
plt.show()

# 计算核密度估计
kde = sns.kdeplot(data, shade=False)
x = kde.get_lines()[0].get_xdata()
y = kde.get_lines()[0].get_ydata()

# 找到峰值
peaks, _ = find_peaks(y)
peak_heights = y[peaks]

# 计算FWHM
fwhm_values = []
for peak, height in zip(peaks, peak_heights):
    half_height = height / 2
    left_idx = np.where(y[:peak] < half_height)[0][-1]
    right_idx = np.where(y[peak:] < half_height)[0][0] + peak
    fwhm = x[right_idx] - x[left_idx]
    fwhm_values.append(fwhm)

print("FWHM values:", fwhm_values)