[Rust] iterators

🚨 현재 공부하고 있는 내용으로 틀린 내용이 있을 수 있습니다.

iterator 는 vector, 배열 등에서 반복적으로 무언가를 수행할 때 용이하게 사용할 수 있는데, 종류로는 크게 3가지가 있다.

1. .iter() -> reference의 형태로 iterator 생성
2. .iter_mut() -> mut ref의 형태로 iterator 생성
3. .into_iter() -> 실제 value로 iterator 생성

예시를 보는 것이 훨씬 이해하기 쉽다.

예시

---

map()

이해를 돕기 위해 map()에 대해 먼저 살펴보자.

core::iter::traits::iterator::Iterator
pub fn map<B, F>(self, f: F) -> Map<Self, F>
where
    Self: Sized,
    F: FnMut(Self::Item) -> B,

map() 은 반복문이랑 비슷하게 생각할 수 있는데, iterator 뒤에 사용하고 FnMut Closure를 인자로 받아서 iterator를 리턴하는 함수다.

collect

그 다음으로 살펴볼 것은 collect 다.

core::iter::traits::iterator::Iterator
pub fn collect<B>(self) -> B
where
    B: FromIterator<Self::Item>,
    Self: Sized,

collect 는 iterator를 받아서 다른 타입으로 바꿔주는 아주 유용한 함수다. collect 뒤에 바꾸고 싶은 타입을 적어주면 된다.

진짜 예제

필요한 것은 모두 봤으니 이제 예제를 보도록 하자.

fn main() {
    let vector1 = vec![1, 2, 3]; 
    let vector1_a = vector1.iter().map(|x| x + 1).collect::<Vec<i32>>();
    let vector1_b = vector1.into_iter().map(|x| x * 10).collect::<Vec<i32>>();

    let mut vector2 = vec![10, 20, 30]; // we will use .iter_mut() on this one
    vector2.iter_mut().for_each(|x| *x +=100);

    println!("{:?}", vector1_a);
    println!("{:?}", vector2);
    println!("{:?}", vector1_b);
}

첫 번째 vector1에 .iter() 를 사용해서 map에 전달하고, map은 closure를 이용해 iterator 를 리턴한다. 생성된 iterator를 collector가 받아서 Vec 타입으로 새로운 vector를 만들어 vector1_a에 할당한다. 이 때 .iter() 는 reference의 형태로 iterator를 만들기 때문에 기존에 사용한 vector1은 그대로 살아있다.

이제 두 번째를 보면 첫 번째와 모든 것이 같지만 .into_iter() 를 사용했다는 것만 다르다. .into_iter() 는 기존에 사용한 vector1을 소비하기 때문에 .into_iter()를 사용하고 나면 더 이상 vector1을 사용할 수 없다.

마지막으로 iter_mut() 를 보면 마찬가지로 ref 형태로 iterator를 만들기 때문에 vector2는 그대로 살아있다. 여기서는 map() 대신 for_each 를 사용했는데, for_each 는 collect 를 사용할 필요 없다는 장점이 있다.

출처 및 참고자료

---

1. https://stackoverflow.com/questions/32872013/does-rust-have-a-way-to-apply-a-function-method-to-each-element-in-an-array-or-v
2. https://dhghomon.github.io/easy_rust/Chapter_36.html?highlight=Iterator#iterators
3. https://youtu.be/uHzakmeaVL8