使用时间增量来正确地对齐图上的峰值

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

# 生成示例信号
np.random.seed(0)
t = np.linspace(0, 1, 1000, endpoint=False)
signal = np.sin(2 * np.pi * 5 * t) + np.random.normal(0, 0.5, len(t))

# 峰值检测
peaks, _ = find_peaks(signal, height=0)
peak_times = t[peaks]

# 对齐峰值
aligned_signal = np.zeros((len(peaks), 100))
for i, peak_time in enumerate(peak_times):
    start_idx = int(peak_time * 10) - 50
    aligned_signal[i] = signal[start_idx:start_idx + 100]

# 可视化结果
plt.figure(figsize=(12, 6))
plt.subplot(2, 1, 1)
plt.plot(t, signal)
plt.plot(peak_times, signal[peaks], "x", label="Peaks")
plt.legend()
plt.subplot(2, 1, 2)
for i in range(len(aligned_signal)):
    plt.plot(aligned_signal[i] + i * 0.5)
plt.show()