Skip to content

Instantly share code, notes, and snippets.

@ayoisaiah
Last active January 30, 2023 12:30
Show Gist options
  • Star 6 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ayoisaiah/185fec1ca98ce44fca1308753182ff2b to your computer and use it in GitHub Desktop.
Save ayoisaiah/185fec1ca98ce44fca1308753182ff2b to your computer and use it in GitHub Desktop.
Find Mean, Median and Mode in Rust
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));
}
@paulstrrr777
Copy link

paulstrrr777 commented Jan 30, 2023

Hi! IMHO median calculates with rounding.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment