[Rust] itertools

#Rust

itertools는 기본 Iter 타입을 보조하는 형태의 확장형 기능이다.
std iterator는 map, filter, 단순한 집계 같은 기본적인 기능은 제공하지만, 복잡한 데이터 처리 기능을 구현하려면 손이 가는 부분이 분명 존재한다.

itertools는 그런 부분들에 대해서 자주 쓰는 패턴들을 추가 구현해놓은 라이브러리다.

설치 및 기본 사용법

설치는 간단하다.

cargo add itertools

사용법도 간단한 편이다.
Itertools라는 트레잇을 use하기만 해도 현재 컨텍스트에서 관련 메서드들이 Iter 객체들에 자동 바인딩된다.

따로 뭘 해야할 필요는 없다.

잡다한 기능이 매우 많은 편인데, 자주 사용할만한 기능들 몇가지만 소개해보겠다.

combination은 현재 컬렉션에서 N개로 이루어지는 조합을 생성해준다.
수학이나 문제풀이할때 나오는 그 조합이 맞다.

조금 자주 사용할만한 기능이다. 모든 데이터가 키 단위로 정렬되어있다는 가정 하에서, 그루핑을 시도한다.

이런 데이터가 있으면

다음과 같이 하면 구매자 단위로 그루핑이 가능하다.

group_by 클로저에서 반환한 값이 그룹을 묶는 기준이 된다.

그리고 group iter 튜플 1 필드에서 반환되는 값이 그루핑된 리스트다.

기대한대로 잘 실행이 되었다.

좀 유의할 점은, 키는 같은 키끼리 모여있어야 의도대로 동작한다는 것이다.
아래처럼 키가 퍼져있으면

다른걸로 묶어버린다.

이것도 꽤 자주 사용할만한 기능이다. 컬렉션을 여러개의 청크 단위로 쪼갠다.

이건 3개 길이의 청크로 쪼개는 예제다.

실행해보면

기대한대로 잘렸음을 확인할 수 있다.

.collect::<Vec<_>>의 줄임말용 함수다.

원래 저거 치는게 좀 귀찮은 편인데, 그거에 비해서 비교적 깔끔하고 타이핑하기도 편하다.