Python itertools ライブラリ完全ガイド
Python 標準ライブラリの itertools は、反復処理（イテレーション）を効率的に行うための便利なツールを提供します。本記事では、itertools の主要な機能とその活用方法について詳しく解説します。
1. itertools ライブラリの概要
itertoolsは、イテレータを操作するための関数を提供します。- ループ処理を効率化し、メモリ効率の良いコードを記述するのに役立ちます。
- 組み合わせ・順列の生成、無限イテレータ、フィルタリング機能などを備えています。
インストール方法
itertools は Python の標準ライブラリなので、追加のインストールは不要です。
import itertools
2. 主な機能と使用例
(1) 無限イテレータ（count, cycle, repeat）
import itertools # count: 1 ずつ増加する無限ループ take_5 = list(itertools.islice(itertools.count(10, 2), 5)) print(take_5) # [10, 12, 14, 16, 18] # cycle: 指定したシーケンスを無限ループ cycler = itertools.cycle(["A", "B", "C"]) print([next(cycler) for _ in range(6)]) # ['A', 'B', 'C', 'A', 'B', 'C'] # repeat: 同じ値を繰り返す print(list(itertools.repeat("X", 4))) # ['X', 'X', 'X', 'X']
使用例:
count はインデックス生成に、cycle はラウンドロビン処理に、repeat はデフォルト値の設定に役立ちます。
(2) 順列と組み合わせの生成（permutations, combinations, combinations_with_replacement）
import itertools items = ['A', 'B', 'C'] # 順列（並べ替え） print(list(itertools.permutations(items, 2))) # [('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')] # 組み合わせ（順番は考慮しない） print(list(itertools.combinations(items, 2))) # [('A', 'B'), ('A', 'C'), ('B', 'C')] # 重複を許可する組み合わせ print(list(itertools.combinations_with_replacement(items, 2))) # [('A', 'A'), ('A', 'B'), ('A', 'C'), ('B', 'B'), ('B', 'C'), ('C', 'C')]
使用例: 組み合わせ最適化やパスワードクラッキングのシミュレーションに利用できます。
(3) フィルタリング（filterfalse, dropwhile, takewhile）
import itertools nums = [1, 2, 3, 4, 5, 6] # 条件に合わないものを抽出（偶数を抽出） print(list(itertools.filterfalse(lambda x: x % 2, nums))) # [2, 4, 6] # 条件が True の間スキップ print(list(itertools.dropwhile(lambda x: x < 4, nums))) # [4, 5, 6] # 条件が True の間取得 print(list(itertools.takewhile(lambda x: x < 4, nums))) # [1, 2, 3]
使用例: 条件に基づいてリストから要素を効率的にフィルタリングできます。
(4) 要素のグループ化（groupby）
import itertools data = [("A", 1), ("A", 2), ("B", 3), ("B", 4), ("C", 5)] for key, group in itertools.groupby(data, key=lambda x: x[0]): print(key, list(group))
使用例: ソート済みデータをグループ化し、カテゴリ別に処理するのに役立ちます。
(5) 連結したイテレータの生成（chain）
import itertools list1 = [1, 2, 3] list2 = [4, 5, 6] # 2つのリストを連結 print(list(itertools.chain(list1, list2))) # [1, 2, 3, 4, 5, 6]
使用例: 複数のリストやイテレータを1つにまとめる際に利用できます。
(6) 複数イテレータの直積（product）
import itertools colors = ["red", "blue"] sizes = ["S", "M", "L"] # 直積を生成 print(list(itertools.product(colors, sizes)))
使用例: 全パターンの組み合わせ（例: 商品オプション）を生成する際に活用できます。
(7) 値の累積和（accumulate）
import itertools numbers = [1, 2, 3, 4] print(list(itertools.accumulate(numbers))) # [1, 3, 6, 10]
使用例: データの累積計算（例: 売上合計）に活用できます。
(8) 指定回数イテレータを繰り返す（tee）
import itertools iterator = iter([1, 2, 3]) iter1, iter2 = itertools.tee(iterator, 2) print(list(iter1)) # [1, 2, 3] print(list(iter2)) # [1, 2, 3]
使用例: 同じイテレータを複数回利用する場合に活用できます。
(9) 指定回数の繰り返し（starmap）
import itertools import operator pairs = [(1, 2), (3, 4), (5, 6)] print(list(itertools.starmap(operator.mul, pairs))) # [2, 12, 30]
使用例: 関数をリストの各ペアに適用する際に活用できます。
(10) 指定回数でイテレータを分割（batched）
import itertools # Python 3.10 以降で利用可能 def batched(iterable, n): it = iter(iterable) while batch := tuple(itertools.islice(it, n)): yield batch print(list(batched(range(10), 3))) # [(0, 1, 2), (3, 4, 5), (6, 7, 8), (9,)]
使用例: データを一定サイズごとに分割する際に活用できます。
まとめ
Python の itertools ライブラリは、反復処理を効率的に行うための強力なツールです。無限イテレータ、組み合わせ・順列、フィルタリング、グループ化など、さまざまなシナリオで活用できます。ぜひ、あなたのコードに活かしてみてください！ 🚀
