Python List Comprehensions Cheat Sheet

squares = [x ** 2 for x in range(10)]# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]evens = [x for x in range(20) if x % 2 == 0]pairs = [(x, y) for x in range(3) for y in range(3)]# [(0,0), (0,1), (0,2), (1,0), ...]

# ternary expression -> transforms every elementlabels = ["even" if x % 2 == 0 else "odd" for x in range(5)]# trailing if -> filters which elements are includedevens_only = [x for x in range(10) if x % 2 == 0]# combining bothresult = [x if x > 0 else 0 for x in [-2, -1, 0, 1, 2] if x != 0]

matrix = [[1, 2, 3], [4, 5, 6]]flat = [num for row in matrix for num in row]# [1, 2, 3, 4, 5, 6]transposed = [[row[i] for row in matrix] for i in range(3)]# [[1, 4], [2, 5], [3, 6]]

squares_dict = {x: x ** 2 for x in range(5)}# {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}unique_lengths = {len(word) for word in ["hi", "bye", "ok"]}# {2, 3}swapped = {v: k for k, v in {"a": 1, "b": 2}.items()}# {1: 'a', 2: 'b'}

gen = (x ** 2 for x in range(10))   # parentheses, not bracketsnext(gen)   # 0next(gen)   # 1# memory-efficient for large rangestotal = sum(x ** 2 for x in range(1_000_000))