'Is it possible to split a vector into groups of 10 with iterators?
I have let my_vec = (0..25).collect::<Vec<_>>()
and I would like to split my_vec
into iterators of groups of 10:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9];
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19];
[20, 21, 22, 23, 24, None, None, None, None, None];
Is it possible to do that with iterators in Rust?
Solution 1:[1]
There is no such helper method on the Iterator
trait directly. However, there are two main ways to do it:
- Use the
[T]::chunks()
method (which can be called on aVec<T>
directly). However, it has a minor difference: it won't produceNone
, but the last iteration yields a smaller slice. For always-exact slices that omit the final chunk if it's incomplete, see the[T]::chunks_exact()
method.
let my_vec = (0..25).collect::<Vec<_>>();
for chunk in my_vec.chunks(10) {
println!("{:02?}", chunk);
}
Result:
```none
[00, 01, 02, 03, 04, 05, 06, 07, 08, 09]
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
[20, 21, 22, 23, 24]
```
- Use the
Itertools::chunks()
method from the crateitertools
. This crate extends theIterator
trait from the standard library so thischunks()
method works with all iterators! Note that the usage is slightly more complicated in order to be that general. This has the same behavior as the method described above: in the last iteration, the chunk will be smaller instead of containingNone
s.
extern crate itertools;
use itertools::Itertools;
for chunk in &(0..25).chunks(10) {
println!("{:02?}", chunk.collect::<Vec<_>>());
}
Result:
```none
[00, 01, 02, 03, 04, 05, 06, 07, 08, 09]
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
[20, 21, 22, 23, 24]
```
Solution 2:[2]
You can achieve a similar solution as Lukas Kalbertodt's itertools example using only the standard library:
let my_vec = (0..25).collect::<Vec<_>>();
let mut my_values = my_vec.into_iter().peekable();
while my_values.peek().is_some() {
let chunk: Vec<_> = my_values.by_ref().take(10).collect();
println!("{:?}", chunk);
}
Result:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
[20, 21, 22, 23, 24]
If you don't actually need the Vec
of each chunk, you can omit the collect
call and operate directly on the iterator created by by_ref
.
See also:
Solution 3:[3]
A trick I adapted from Python to get a groupby()
effect like Python's itertools
provides is to combine two or more iterators using .zip()
, .skip()
and .step_by()
. This approach could produce groups of 10, but it would be quite unaesthetic to see in a code base. But if you need small groups, this may be okay.
Rust does have the itertools
crate which has .group_by()
, but that's not always available depending on the situation - for instance, submitting a Rust solution to sites like HackerRank.
fn main()
{
let a = "Hello, World!";
println!("{:?}", a.chars().step_by(2).zip(
a.chars().skip(1).step_by(2)
).collect::<Vec<_>>());
}
Output:
[('H', 'e'), ('l', 'l'), ('o', ','), (' ', 'W'), ('o', 'r'), ('l', 'd')]
Where n
is the number of zipped iterators, the value for .step_by
will be n
for all iterators, and the value for .skip()
will depend on the position of the iterator in the chain; the first iterator will skip 0, the next 1, etc.
You can chain any number of iterators this way, but it starts looking a bit complicated and ugly after the third. In that case, maybe consider just collecting the iterator into a Vec
and using its .chunks()
method. Or if you can, use the itertools
crate.
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
Solution | Source |
---|---|
Solution 1 | Brendano257 |
Solution 2 | Shepmaster |
Solution 3 |