Nested Loops in Comprehensions
- Pattern:
[expr for a in A for b in B]
A = [1, 2]
B = ['x', 'y']
pairs = [(a, b) for a in A for b in B]
print(pairs)
- Flatten one level of a 2D list:
matrix = [[1, 2], [3, 4, 5], [6]]
flat = [x for row in matrix for x in row]
print(flat)