Last active
January 30, 2023 12:30
-
-
Save ayoisaiah/185fec1ca98ce44fca1308753182ff2b to your computer and use it in GitHub Desktop.
Find Mean, Median and Mode in Rust
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
use std::collections::HashMap; | |
fn mean(numbers: &Vec<i32>) -> f32 { | |
let sum: i32 = numbers.iter().sum(); | |
sum as f32 / numbers.len() as f32 | |
} | |
fn median(numbers: &mut Vec<i32>) -> i32 { | |
numbers.sort(); | |
let mid = numbers.len() / 2; | |
if numbers.len() % 2 == 0 { | |
mean(&vec![numbers[mid - 1], numbers[mid]]) as i32 | |
} else { | |
numbers[mid] | |
} | |
} | |
fn mode(numbers: &Vec<i32>) -> Vec<i32> { | |
let mut map = HashMap::new(); | |
for integer in numbers { | |
let count = map.entry(integer).or_insert(0); | |
*count += 1; | |
} | |
let max_value = map.values().cloned().max().unwrap_or(0); | |
map.into_iter() | |
.filter(|&(_, v)| v == max_value) | |
.map(|(&k, _)| k) | |
.collect() | |
} | |
fn main() { | |
let mut data: Vec<i32> = vec![ | |
1, | |
1, | |
2, | |
3, | |
26, | |
34, | |
36, | |
36, | |
38, | |
43, | |
43, | |
43, | |
43, | |
54, | |
76, | |
328, | |
378, | |
2928, | |
]; | |
println!("MEAN: {}", mean(&data)); | |
println!("MEDIAN: {}", median(&mut data)); | |
println!("MODE: {:?}", mode(&data)); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi! IMHO median calculates with rounding.