🔗 Python Tutorial — Itertools Module

Introduction 🌟

The itertools module provides powerful tools for working with iterators. It helps you loop efficiently, generate combinations, create infinite iterators, and build complex iteration logic.

Note

💡 itertools = “Iterator Tools”
💡 Highly optimized → super fast
💡 Used in algorithms, data processing, AI, simulations, and combinatorics

1. Importing itertools 🧱

import_itertools.py

import itertools

2. Infinite Iterators ♾️

count() — Infinite Counting

count_example.py

for num in itertools.count(start=10, step=2):
    print(num)
    if num > 20:
        break

✔ Generates: 10, 12, 14, ...

cycle() — Repeat a Sequence Forever

cycle_example.py

for item in itertools.cycle(["A", "B", "C"]):
    print(item)
    break

✔ Cycles through values infinitely

repeat() — Repeat a Value

repeat_example.py

for x in itertools.repeat("Hi", 3):
    print(x)

✔ Useful for filling data

3. Combinatoric Iterators 🎯

product() — Cartesian Product

product_example.py

for p in itertools.product([1, 2], ["A", "B"]):
    print(p)

✔ Output: (1, 'A'), (1, 'B'), (2, 'A'), (2, 'B')

permutations()

permutations_example.py

for p in itertools.permutations([1, 2, 3], 2):
    print(p)

✔ Order matters → (1,2), (2,1)…

combinations()

combinations_example.py

for c in itertools.combinations([1, 2, 3], 2):
    print(c)

✔ Order doesn't matter → (1,2), (1,3), (2,3)

combinations_with_replacement()

cwr_example.py

for c in itertools.combinations_with_replacement([1, 2], 2):
    print(c)

✔ Ex: (1,1), (1,2), (2,2)

4. Utility Iterators 🧰

accumulate() — Running Totals

accumulate_example.py

import itertools, operator

nums = [1, 2, 3, 4]
print(list(itertools.accumulate(nums)))
print(list(itertools.accumulate(nums, operator.mul)))

✔ Default: sum
✔ With operator.mul → running product

chain() — Join Iterables

chain_example.py

for item in itertools.chain([1, 2], ["A", "B"]):
    print(item)

✔ Merges multiple iterables

chain.from_iterable()

chain_from_iterable.py

nested = [[1, 2], [3, 4]]
print(list(itertools.chain.from_iterable(nested)))

compress() — Filter by Selector

compress_example.py

items = ["A", "B", "C", "D"]
selectors = [1, 0, 1, 0]

print(list(itertools.compress(items, selectors)))

✔ Output: ["A", "C"]

dropwhile() & takewhile()

drop_take_while.py

nums = [1, 2, 3, 4, 1]

print(list(itertools.dropwhile(lambda x: x < 3, nums)))
print(list(itertools.takewhile(lambda x: x < 3, nums)))

✔ dropwhile → skip until condition false
✔ takewhile → take while condition true

filterfalse()

filterfalse_example.py

print(list(itertools.filterfalse(lambda x: x % 2 == 0, [1,2,3,4])))

✔ Opposite of filter()

5. Grouping Data — groupby() 📊

groupby_example.py

data = [("A", 1), ("A", 2), ("B", 1), ("B", 3)]

for key, group in itertools.groupby(data, lambda x: x[0]):
    print(key, list(group))

Note

💡 groupby requires data to be sorted by the key

6. Real-World Example — Password Generator 🔐

password_itertools.py

import itertools, string

chars = string.ascii_letters + string.digits
passwords = itertools.product(chars, repeat=3)

for pwd in passwords:
    print("".join(pwd))
    break

✔ Generates password combinations

7. Real-World Example — Flatten Nested Lists 📥

flatten_example.py

nested = [[1, 2], [3, 4, 5]]
flat = list(itertools.chain.from_iterable(nested))
print(flat)

8. Itertools Cheat Sheet 📘

CategoryFunctions
Infinite Iteratorscount, cycle, repeat
Combinatoricsproduct, permutations, combinations
Filteringcompress, dropwhile, takewhile, filterfalse
Joiningchain, chain.from_iterable
Accumulationaccumulate
Groupinggroupby

Best Practices 💡

  • ✔ Use product, combinations for combinatorics
  • ✔ Use infinite iterators carefully → always add a break
  • ✔ Prefer chain for merging lists instead of “+”
  • ✔ Use groupby after sorting data
  • ✔ itertools works best with generators → memory efficient

Conclusion 🎉

>>“itertools transforms simple loops into elegant, powerful, high-performance iteration pipelines.” ✨

You now fully understand the itertools module! Want the next topic? Try Functools, Statistics, Shutil, or Pathlib. Just tell me! 😊