Function create_f64_array_with_seed

arrow/src/util/bench_util.rs:861–873

Creates a random f64 array of a given size and nan-value density based on a given seed

(size: usize, nan_density: f32, seed: u64)

/// Creates a random f64 array of a given size and nan-value density based on a given seed
pub fn create_f64_array_with_seed(size: usize, nan_density: f32, seed: u64) -> Float64Array {
    let mut rng = StdRng::seed_from_u64(seed);

    (0..size)
        .map(|_| {
            if rng.random::<f32>() < nan_density {
                Some(f64::NAN)
            } else {
                Some(rng.random())
            }
        })
        .collect()
}