Advanced Python Iteration Techniques with Practical Examples

Python's advanced iteration techniques can make code more concise, efficient, and readable, covering list, dict, and set comprehensions, the zip and enumerate functions, various itertools utilities, and generator expressions.

Example 1: List comprehension to compute squares

numbers = [1, 2, 3, 4, 5]
squares = [num**2 for num in numbers]
print(squares)  # 输出 [1, 4, 9, 16, 25]

Example 2: Dictionary comprehension to create a mapping

keys = ['a', 'b', 'c']
values = [1, 2, 3]
mapping = {key: value for key, value in zip(keys, values)}
print(mapping)  # 输出 {'a': 1, 'b': 2, 'c': 3}

Example 3: Set comprehension to remove duplicates

items = [1, 2, 2, 3, 4, 4, 5]
unique_items = {item for item in items}
print(unique_items)  # 输出 {1, 2, 3, 4, 5}

Example 4: Using zip to iterate over multiple sequences in parallel

names = ["Alice", "Bob", "Charlie"]
ages = [25, 30, 35]
for name, age in zip(names, ages):
    print(f"{name} is {age} years old.")

Example 5: Using enumerate to get index and value

fruits = ['apple', 'banana', 'cherry']
for index, fruit in enumerate(fruits):
    print(f"Item {index}: {fruit}")

Example 6: Using itertools.chain to flatten a nested list

from itertools import chain
nested_list = [[1, 2], [3, 4], [5]]
flat_list = list(chain(*nested_list))
print(flat_list)  # 输出 [1, 2, 3, 4, 5]

Example 7: Using itertools.groupby to group data by a key

from itertools import groupby
data = [(1, 'A'), (1, 'B'), (2, 'C'), (2, 'D')]
sorted_data = sorted(data, key=lambda x: x[0])
grouped_data = {k: list(v) for k, v in groupby(sorted_data, lambda x: x[0])}
print(grouped_data)  # 输出 {1: [(1, 'A'), (1, 'B')], 2: [(2, 'C'), (2, 'D')]}

Example 8: Using itertools.permutations to generate permutations

from itertools import permutations
letters = ['A', 'B', 'C']
permutations_list = list(permutations(letters))
print(permutations_list)

Example 9: Using itertools.product to generate Cartesian product

from itertools import product
colors = ['Red', 'Blue']
sizes = ['Small', 'Large']
combinations = list(product(colors, sizes))
print(combinations)

Example 10: Using a generator expression to save memory

large_numbers = (num for num in range(1000000))
sum_of_large_numbers = sum(large_numbers)
print(sum_of_large_numbers)

These examples demonstrate how advanced iteration techniques in Python can help you write more efficient and concise code; applying them thoughtfully can improve performance while maintaining readability.