group-by 1.0.0

A group_by function
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use std::collections::HashMap;
use std::hash::Hash;

/// Group the iterator by the result of `f`. Similar to Ruby's `group_by`.
pub fn group_by<T, K, F>(x: T, f: F) -> HashMap<K, Vec<T::Item>>
    where T: IntoIterator,
          F: Fn(&T::Item) -> K,
          K: Eq + Hash
{
    let mut map = HashMap::new();
    for item in x {
        let hash = f(&item);
        map.entry(hash).or_insert(vec![]).push(item);
    }
    map
}

#[test]
fn it_works() {
    let map = group_by(1..7, |i| i%3);
    let expected = vec![
        (0, vec![3, 6]),
        (1, vec![1, 4]),
        (2, vec![2, 5]),
    ].into_iter().collect();
    assert_eq!(map, expected);
}