#data-science #signal #spectrum #data-points #prominence

find_peaks

Find peaks that match criteria in 1D data

6 releases

0.1.5 May 18, 2022
0.1.4 May 2, 2022
0.1.3 Mar 6, 2022
0.1.2 Oct 8, 2020
0.1.1 Sep 24, 2020

#499 in Math

Download history 28/week @ 2023-11-21 77/week @ 2023-11-28 176/week @ 2023-12-05 75/week @ 2023-12-12 68/week @ 2023-12-19 49/week @ 2023-12-26 82/week @ 2024-01-02 64/week @ 2024-01-09 30/week @ 2024-01-16 45/week @ 2024-01-23 41/week @ 2024-01-30 77/week @ 2024-02-06 45/week @ 2024-02-13 111/week @ 2024-02-20 35/week @ 2024-02-27 21/week @ 2024-03-05

219 downloads per month
Used in mathbox

MIT license

53KB
547 lines

Description

Find a filtered subset of local maxima in 1D slice of data.

The functionality implemented here is might be familiar to anyone using MATLAB's findpeaks, or Python's scipy.signal.find_peaks.

Arguably, the most useful feature in this package is filtering peaks through prominence. This parameter allows you to get the subset of local maxima that optically look like peaks even in noisy data.

Filtering conditions that can be set are:

  • prominence,
  • height,
  • the absolute value of the difference between neighboring data points,
  • number of plateau points a peak can have,
  • distance between peaks.

All parameters can be specified by minimum and maximum bound.

Elements of the data slice need not be of a specific type, as long as they implement a few traits (for cloning, subtraction, comparison).

Example

Copied from examples/spectrum.rs.

use find_peaks::PeakFinder;

use std::fs::File;
use std::io::prelude::*;

fn read_file(path: &str) -> std::io::Result<String> {
    let mut file = File::open(path)?;
    let mut contents = String::new();
    file.read_to_string(&mut contents)?;
    Ok(contents)
}

fn main() -> () {
    let data: Vec<f64> = read_file("data/spectrum.dat").expect("File not read!").as_str()
        .split_whitespace()
        .map(|x| x.parse::<f64>().unwrap())
        .collect();
    
    let mut fp = PeakFinder::new(&data);
    fp.with_min_prominence(200.);
    fp.with_min_height(0.);

    let peaks = fp.find_peaks();
    for p in peaks {
        println!("{} {}", p.middle_position(), p.height.unwrap());
    }
}

The result visualized:

No runtime deps