Struct itertools::structs::GroupingMap

source · 
pub struct GroupingMap<I> { /* private fields */ }
Expand description

GroupingMap is an intermediate struct for efficient group-and-fold operations. It groups elements by their key and at the same time fold each group using some aggregating operation.

No method on this struct performs temporary allocations.

Implementations§

source§

impl<I, K, V> GroupingMap<I>
where I: Iterator<Item = (K, V)>, K: Hash + Eq,

source

pub fn aggregate<FO, R>(self, operation: FO) -> HashMap<K, R>
where FO: FnMut(Option<R>, &K, V) -> Option<R>,

This is the generic way to perform any operation on a GroupingMap. It’s suggested to use this method only to implement custom operations when the already provided ones are not enough.

Groups elements from the GroupingMap source by key and applies operation to the elements of each group sequentially, passing the previously accumulated value, a reference to the key and the current element as arguments, and stores the results in an HashMap.

The operation function is invoked on each element with the following parameters:

  • the current value of the accumulator of the group if there is currently one;
  • a reference to the key of the group this element belongs to;
  • the element from the source being aggregated;

If operation returns Some(element) then the accumulator is updated with element, otherwise the previous accumulation is discarded.

Return a HashMap associating the key of each group with the result of aggregation of that group’s elements. If the aggregation of the last element of a group discards the accumulator then there won’t be an entry associated to that group’s key.

use itertools::Itertools;

let data = vec![2, 8, 5, 7, 9, 0, 4, 10];
let lookup = data.into_iter()
    .into_grouping_map_by(|&n| n % 4)
    .aggregate(|acc, _key, val| {
        if val == 0 || val == 10 {
            None
        } else {
            Some(acc.unwrap_or(0) + val)
        }
    });

assert_eq!(lookup[&0], 4);        // 0 resets the accumulator so only 4 is summed
assert_eq!(lookup[&1], 5 + 9);
assert_eq!(lookup.get(&2), None); // 10 resets the accumulator and nothing is summed afterward
assert_eq!(lookup[&3], 7);
assert_eq!(lookup.len(), 3);      // The final keys are only 0, 1 and 2
source

pub fn fold_with<FI, FO, R>(self, init: FI, operation: FO) -> HashMap<K, R>
where FI: FnMut(&K, &V) -> R, FO: FnMut(R, &K, V) -> R,

Groups elements from the GroupingMap source by key and applies operation to the elements of each group sequentially, passing the previously accumulated value, a reference to the key and the current element as arguments, and stores the results in a new map.

init is called to obtain the initial value of each accumulator.

operation is a function that is invoked on each element with the following parameters:

  • the current value of the accumulator of the group;
  • a reference to the key of the group this element belongs to;
  • the element from the source being accumulated.

Return a HashMap associating the key of each group with the result of folding that group’s elements.

use itertools::Itertools;

#[derive(Debug, Default)]
struct Accumulator {
  acc: usize,
}

let lookup = (1..=7)
    .into_grouping_map_by(|&n| n % 3)
    .fold_with(|_key, _val| Default::default(), |Accumulator { acc }, _key, val| {
        let acc = acc + val;
        Accumulator { acc }
     });

assert_eq!(lookup[&0].acc, 3 + 6);
assert_eq!(lookup[&1].acc, 1 + 4 + 7);
assert_eq!(lookup[&2].acc, 2 + 5);
assert_eq!(lookup.len(), 3);
source

pub fn fold<FO, R>(self, init: R, operation: FO) -> HashMap<K, R>
where R: Clone, FO: FnMut(R, &K, V) -> R,

Groups elements from the GroupingMap source by key and applies operation to the elements of each group sequentially, passing the previously accumulated value, a reference to the key and the current element as arguments, and stores the results in a new map.

init is the value from which will be cloned the initial value of each accumulator.

operation is a function that is invoked on each element with the following parameters:

  • the current value of the accumulator of the group;
  • a reference to the key of the group this element belongs to;
  • the element from the source being accumulated.

Return a HashMap associating the key of each group with the result of folding that group’s elements.

use itertools::Itertools;

let lookup = (1..=7)
    .into_grouping_map_by(|&n| n % 3)
    .fold(0, |acc, _key, val| acc + val);

assert_eq!(lookup[&0], 3 + 6);
assert_eq!(lookup[&1], 1 + 4 + 7);
assert_eq!(lookup[&2], 2 + 5);
assert_eq!(lookup.len(), 3);
source

pub fn fold_first<FO>(self, operation: FO) -> HashMap<K, V>
where FO: FnMut(V, &K, V) -> V,

Groups elements from the GroupingMap source by key and applies operation to the elements of each group sequentially, passing the previously accumulated value, a reference to the key and the current element as arguments, and stores the results in a new map.

This is similar to fold but the initial value of the accumulator is the first element of the group.

operation is a function that is invoked on each element with the following parameters:

  • the current value of the accumulator of the group;
  • a reference to the key of the group this element belongs to;
  • the element from the source being accumulated.

Return a HashMap associating the key of each group with the result of folding that group’s elements.

use itertools::Itertools;

let lookup = (1..=7)
    .into_grouping_map_by(|&n| n % 3)
    .fold_first(|acc, _key, val| acc + val);

assert_eq!(lookup[&0], 3 + 6);
assert_eq!(lookup[&1], 1 + 4 + 7);
assert_eq!(lookup[&2], 2 + 5);
assert_eq!(lookup.len(), 3);
source

pub fn collect<C>(self) -> HashMap<K, C>
where C: Default + Extend<V>,

Groups elements from the GroupingMap source by key and collects the elements of each group in an instance of C. The iteration order is preserved when inserting elements.

Return a HashMap associating the key of each group with the collection containing that group’s elements.

use itertools::Itertools;
use std::collections::HashSet;

let lookup = vec![0, 1, 2, 3, 4, 5, 6, 2, 3, 6].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .collect::<HashSet<_>>();

assert_eq!(lookup[&0], vec![0, 3, 6].into_iter().collect::<HashSet<_>>());
assert_eq!(lookup[&1], vec![1, 4].into_iter().collect::<HashSet<_>>());
assert_eq!(lookup[&2], vec![2, 5].into_iter().collect::<HashSet<_>>());
assert_eq!(lookup.len(), 3);
source

pub fn max(self) -> HashMap<K, V>
where V: Ord,

Groups elements from the GroupingMap source by key and finds the maximum of each group.

If several elements are equally maximum, the last element is picked.

Returns a HashMap associating the key of each group with the maximum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .max();

assert_eq!(lookup[&0], 12);
assert_eq!(lookup[&1], 7);
assert_eq!(lookup[&2], 8);
assert_eq!(lookup.len(), 3);
source

pub fn max_by<F>(self, compare: F) -> HashMap<K, V>
where F: FnMut(&K, &V, &V) -> Ordering,

Groups elements from the GroupingMap source by key and finds the maximum of each group with respect to the specified comparison function.

If several elements are equally maximum, the last element is picked.

Returns a HashMap associating the key of each group with the maximum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .max_by(|_key, x, y| y.cmp(x));

assert_eq!(lookup[&0], 3);
assert_eq!(lookup[&1], 1);
assert_eq!(lookup[&2], 5);
assert_eq!(lookup.len(), 3);
source

pub fn max_by_key<F, CK>(self, f: F) -> HashMap<K, V>
where F: FnMut(&K, &V) -> CK, CK: Ord,

Groups elements from the GroupingMap source by key and finds the element of each group that gives the maximum from the specified function.

If several elements are equally maximum, the last element is picked.

Returns a HashMap associating the key of each group with the maximum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .max_by_key(|_key, &val| val % 4);

assert_eq!(lookup[&0], 3);
assert_eq!(lookup[&1], 7);
assert_eq!(lookup[&2], 5);
assert_eq!(lookup.len(), 3);
source

pub fn min(self) -> HashMap<K, V>
where V: Ord,

Groups elements from the GroupingMap source by key and finds the minimum of each group.

If several elements are equally minimum, the first element is picked.

Returns a HashMap associating the key of each group with the minimum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .min();

assert_eq!(lookup[&0], 3);
assert_eq!(lookup[&1], 1);
assert_eq!(lookup[&2], 5);
assert_eq!(lookup.len(), 3);
source

pub fn min_by<F>(self, compare: F) -> HashMap<K, V>
where F: FnMut(&K, &V, &V) -> Ordering,

Groups elements from the GroupingMap source by key and finds the minimum of each group with respect to the specified comparison function.

If several elements are equally minimum, the first element is picked.

Returns a HashMap associating the key of each group with the minimum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .min_by(|_key, x, y| y.cmp(x));

assert_eq!(lookup[&0], 12);
assert_eq!(lookup[&1], 7);
assert_eq!(lookup[&2], 8);
assert_eq!(lookup.len(), 3);
source

pub fn min_by_key<F, CK>(self, f: F) -> HashMap<K, V>
where F: FnMut(&K, &V) -> CK, CK: Ord,

Groups elements from the GroupingMap source by key and finds the element of each group that gives the minimum from the specified function.

If several elements are equally minimum, the first element is picked.

Returns a HashMap associating the key of each group with the minimum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .min_by_key(|_key, &val| val % 4);

assert_eq!(lookup[&0], 12);
assert_eq!(lookup[&1], 4);
assert_eq!(lookup[&2], 8);
assert_eq!(lookup.len(), 3);
source

pub fn minmax(self) -> HashMap<K, MinMaxResult<V>>
where V: Ord,

Groups elements from the GroupingMap source by key and find the maximum and minimum of each group.

If several elements are equally maximum, the last element is picked. If several elements are equally minimum, the first element is picked.

See .minmax() for the non-grouping version.

Differences from the non grouping version:

  • It never produces a MinMaxResult::NoElements
  • It doesn’t have any speedup

Returns a HashMap associating the key of each group with the minimum and maximum of that group’s elements.

use itertools::Itertools;
use itertools::MinMaxResult::{OneElement, MinMax};

let lookup = vec![1, 3, 4, 5, 7, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .minmax();

assert_eq!(lookup[&0], MinMax(3, 12));
assert_eq!(lookup[&1], MinMax(1, 7));
assert_eq!(lookup[&2], OneElement(5));
assert_eq!(lookup.len(), 3);
source

pub fn minmax_by<F>(self, compare: F) -> HashMap<K, MinMaxResult<V>>
where F: FnMut(&K, &V, &V) -> Ordering,

Groups elements from the GroupingMap source by key and find the maximum and minimum of each group with respect to the specified comparison function.

If several elements are equally maximum, the last element is picked. If several elements are equally minimum, the first element is picked.

It has the same differences from the non-grouping version as minmax.

Returns a HashMap associating the key of each group with the minimum and maximum of that group’s elements.

use itertools::Itertools;
use itertools::MinMaxResult::{OneElement, MinMax};

let lookup = vec![1, 3, 4, 5, 7, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .minmax_by(|_key, x, y| y.cmp(x));

assert_eq!(lookup[&0], MinMax(12, 3));
assert_eq!(lookup[&1], MinMax(7, 1));
assert_eq!(lookup[&2], OneElement(5));
assert_eq!(lookup.len(), 3);
source

pub fn minmax_by_key<F, CK>(self, f: F) -> HashMap<K, MinMaxResult<V>>
where F: FnMut(&K, &V) -> CK, CK: Ord,

Groups elements from the GroupingMap source by key and find the elements of each group that gives the minimum and maximum from the specified function.

If several elements are equally maximum, the last element is picked. If several elements are equally minimum, the first element is picked.

It has the same differences from the non-grouping version as minmax.

Returns a HashMap associating the key of each group with the minimum and maximum of that group’s elements.

use itertools::Itertools;
use itertools::MinMaxResult::{OneElement, MinMax};

let lookup = vec![1, 3, 4, 5, 7, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .minmax_by_key(|_key, &val| val % 4);

assert_eq!(lookup[&0], MinMax(12, 3));
assert_eq!(lookup[&1], MinMax(4, 7));
assert_eq!(lookup[&2], OneElement(5));
assert_eq!(lookup.len(), 3);
source

pub fn sum(self) -> HashMap<K, V>
where V: Add<V, Output = V>,

Groups elements from the GroupingMap source by key and sums them.

This is just a shorthand for self.fold_first(|acc, _, val| acc + val). It is more limited than Iterator::sum since it doesn’t use the Sum trait.

Returns a HashMap associating the key of each group with the sum of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .sum();

assert_eq!(lookup[&0], 3 + 9 + 12);
assert_eq!(lookup[&1], 1 + 4 + 7);
assert_eq!(lookup[&2], 5 + 8);
assert_eq!(lookup.len(), 3);
source

pub fn product(self) -> HashMap<K, V>
where V: Mul<V, Output = V>,

Groups elements from the GroupingMap source by key and multiply them.

This is just a shorthand for self.fold_first(|acc, _, val| acc * val). It is more limited than Iterator::product since it doesn’t use the Product trait.

Returns a HashMap associating the key of each group with the product of that group’s elements.

use itertools::Itertools;

let lookup = vec![1, 3, 4, 5, 7, 8, 9, 12].into_iter()
    .into_grouping_map_by(|&n| n % 3)
    .product();

assert_eq!(lookup[&0], 3 * 9 * 12);
assert_eq!(lookup[&1], 1 * 4 * 7);
assert_eq!(lookup[&2], 5 * 8);
assert_eq!(lookup.len(), 3);

Trait Implementations§

source§

impl<I: Clone> Clone for GroupingMap<I>

source§

fn clone(&self) -> GroupingMap<I>

Returns a copy of the value. Read more
1.0.0 · source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
source§

impl<I: Debug> Debug for GroupingMap<I>

source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

Auto Trait Implementations§

§

impl<I> Freeze for GroupingMap<I>
where I: Freeze,

§

impl<I> RefUnwindSafe for GroupingMap<I>
where I: RefUnwindSafe,

§

impl<I> Send for GroupingMap<I>
where I: Send,

§

impl<I> Sync for GroupingMap<I>
where I: Sync,

§

impl<I> Unpin for GroupingMap<I>
where I: Unpin,

§

impl<I> UnwindSafe for GroupingMap<I>
where I: UnwindSafe,

Blanket Implementations§

source§

impl<T> Any for T
where T: 'static + ?Sized,

source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
source§

impl<T> Borrow<T> for T
where T: ?Sized,

source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
source§

impl<T> CloneToUninit for T
where T: Clone,

source§

default unsafe fn clone_to_uninit(&self, dst: *mut T)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dst. Read more
source§

impl<T> From<T> for T

source§

fn from(t: T) -> T

Returns the argument unchanged.

source§

impl<T, U> Into<U> for T
where U: From<T>,

source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

source§

impl<T> ToOwned for T
where T: Clone,

§

type Owned = T

The resulting type after obtaining ownership.
source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

§

type Error = Infallible

The type returned in the event of a conversion error.
source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.