How do I create an iterator over an array which takes values from index i and i+1?

Question

I created a small example that runs in the Rust Playground:

#[derive(Clone)]
pub struct Person {
    pub firstname: Vec<u8>,
    pub surname: Vec<u8>,
}

fn main() {
    let p0 = Person {
        firstname: vec![0u8; 5],
        surname: vec![1u8; 5],
    };
    let p1 = Person {
        firstname: vec![2u8; 7],
        surname: vec![3u8; 2],
    };
    let p2 = Person {
        firstname: vec![4u8; 8],
        surname: vec![5u8; 8],
    };
    let p3 = Person {
        firstname: vec![6u8; 3],
        surname: vec![7u8; 1],
    };

    let people = [p0, p1, p2, p3];

    for i in 0..people.len() {
        if i + 1 < people.len() {
            println!(
                "{:?}",
                (people[i].firstname.clone(), people[i + 1].surname.clone())
            )
        }
    }
}

Given the array people, I want to iterate its elements and collect tuples of the first name of the person at index i and the surname of the person at index i+1. This simple for loop does the job, but let's assume that instead of println I would like to pass such tuple into some function f. I can easily do this in this for loop, but I would like to learn whether I can implement that using an iterator iter() (and later apply collect() or fold functions if needed) instead of using the for loop?

Shepmaster · Accepted Answer · 2019-12-09 15:21:22Z

2

I would use slice::windows:

for window in people.windows(2) {
    println!("{:?}", (&window[0].firstname, &window[1].surname))
}

2 Comments

Stargateur Over a year ago

I would add docs.rs/itertools/0.8.2/itertools/… to the suggestion.

Shepmaster Over a year ago

@Stargateur that's the "Are there equivalents..." question in the "see also" section. Since this case starts from an array/slice, I didn't feel like it needed to be directly mentioned.

Stargateur · Accepted Answer · 2019-12-09 15:21:33Z

2

You can combine two features:

Create an array iterator that skips the first elements (iter().skip(1)).
Iterator.zip

#[derive(Clone)]
pub struct Person {
    pub firstname: Vec<u8>,
    pub surname: Vec<u8>,
}

fn main() {
    let p0 = Person {
        firstname: vec![0u8; 5],
        surname: vec![1u8; 5],
    };
    let p1 = Person {
        firstname: vec![2u8; 7],
        surname: vec![3u8; 2],
    };
    let p2 = Person {
        firstname: vec![4u8; 8],
        surname: vec![5u8; 8],
    };
    let p3 = Person {
        firstname: vec![6u8; 3],
        surname: vec![7u8; 1],
    };

    let people = [p0, p1, p2, p3];

    for (i, j) in people.iter().zip(people.iter().skip(1)) {
        println!("{:?}", (&i.firstname, &j.surname));
    }
}

Permalink to the playground

edited Dec 9, 2019 at 15:21

Stargateur

27.1k9 gold badges73 silver badges99 bronze badges

answered Dec 9, 2019 at 15:12

mcarton

30.6k5 gold badges104 silver badges112 bronze badges

Collectives™ on Stack Overflow

How do I create an iterator over an array which takes values from index i and i+1?

2 Answers 2

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related