ecg.py 文件源码

def lomb(data: List[DataPoint],
         low_frequency: float,
         high_frequency: float):
    """
    Lomb–Scargle periodogram implementation
    :param data: List[DataPoint]
    :param high_frequency: float
    :param low_frequency: float
    :return lomb-scargle pgram and frequency values
    """

    time_stamps = np.array([dp.start_time.timestamp() for dp in data])
    samples = np.array([dp.sample for dp in data])
    frequency_range = np.linspace(low_frequency, high_frequency, len(data))
    result = signal.lombscargle(time_stamps, samples, frequency_range)
    return result, frequency_range