컬렉션과 컴프리헨션

nums: list[int] = [1, 2, 3]

nums.append(4)         # [1, 2, 3, 4]
nums.insert(0, 0)      # [0, 1, 2, 3, 4]
nums.remove(2)         # [0, 1, 3, 4]
last = nums.pop()      # 4 반환, 리스트는 [0, 1, 3]

print(len(nums))       # 3
print(nums[0])         # 0
print(3 in nums)       # True

items = [10, 20, 30, 40, 50]

items[1:3]    # [20, 30]    1 이상 3 미만
items[:2]     # [10, 20]    처음부터 2 미만
items[3:]     # [40, 50]    3 이상부터 끝까지
items[-2:]    # [40, 50]    끝에서 2개
items[::2]    # [10, 30, 50]  step 2
items[::-1]   # [50, 40, 30, 20, 10]   뒤집기

[1, 2] + [3, 4]   # [1, 2, 3, 4]
[0] * 5           # [0, 0, 0, 0, 0]

matrix = [[]] * 3
matrix[0].append(1)
print(matrix)
# [[1], [1], [1]]   ← 모두 같은 리스트를 가리킴!

point: tuple[float, float] = (1.0, 2.0)
person: tuple[str, int] = ("커티스", 30)

# tuple 은 보통 unpack 으로 풀어서 씀
name, age = person
print(name, age)   # 커티스 30

# 한 원소 tuple 은 콤마가 필수 — 괄호만으론 부족
single = (42,)     # tuple
not_tuple = (42)   # 그냥 정수

from typing import NamedTuple

class Person(NamedTuple):
    name: str
    age: int

p = Person("커티스", 30)
print(p.name, p.age)   # 커티스 30

# tuple 처럼 unpack 도 됨
name, age = p

user: dict[str, int] = {"id": 1, "age": 30}

print(user["id"])              # 1
print(user.get("nope"))        # None  (없는 키 → 안전)
print(user.get("nope", -1))    # -1    (기본값)

user["name"] = "curtis"        # 추가/수정
del user["age"]                # 삭제
print("name" in user)          # True

for key in user:                # 키만
    print(key)

for key, value in user.items(): # 둘 다
    print(key, value)

for value in user.values():     # 값만
    print(value)

defaults = {"a": 1, "b": 2}
overrides = {"b": 20, "c": 30}

merged = defaults | overrides
print(merged)    # {"a": 1, "b": 20, "c": 30}

unique: set[int] = {1, 2, 3, 2, 1}
print(unique)   # {1, 2, 3}

unique.add(4)      # {1, 2, 3, 4}
unique.discard(2)  # {1, 3, 4}
print(3 in unique) # True

# 빈 set 은 set() — {} 는 빈 dict
empty = set()

a = {1, 2, 3, 4}
b = {3, 4, 5, 6}

a | b   # 합집합  {1, 2, 3, 4, 5, 6}
a & b   # 교집합  {3, 4}
a - b   # 차집합  {1, 2}
a ^ b   # 대칭차  {1, 2, 5, 6}

items = [1, 2, 1, 3, 2, 4]
unique = list(dict.fromkeys(items))
print(unique)   # [1, 2, 3, 4]

# [표현식 for 변수 in 이터러블]

squares = [x ** 2 for x in range(5)]
# [0, 1, 4, 9, 16]

squares = []
for x in range(5):
    squares.append(x ** 2)

evens = [x for x in range(10) if x % 2 == 0]
# [0, 2, 4, 6, 8]

# 짝수만 골라서 제곱
result = [x ** 2 for x in range(10) if x % 2 == 0]
# [0, 4, 16, 36, 64]

labels = ["even" if x % 2 == 0 else "odd" for x in range(5)]
# ['even', 'odd', 'even', 'odd', 'even']

matrix = [[0 for _ in range(3)] for _ in range(3)]
# [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

square_map = {x: x ** 2 for x in range(5)}
# {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

# 키-값 swap
user = {"id": 1, "name": "curtis"}
swapped = {v: k for k, v in user.items()}
# {1: "id", "curtis": "name"}

unique_lengths = {len(w) for w in ["a", "bb", "cc", "ddd"]}
# {1, 2, 3}

pairs = [(x, y) for x in [1, 2] for y in ['a', 'b']]
# [(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')]

pairs = []
for x in [1, 2]:
    for y in ['a', 'b']:
        pairs.append((x, y))

gen = (x ** 2 for x in range(1_000_000))
print(gen)   # <generator object ...>

total = sum(x ** 2 for x in range(1_000_000))
# 함수에 바로 넣으면 ()를 생략할 수 있음

# 한 줄로 가능은 한데, 읽기 어려움
result = [transform(x) for x in items if validate(x) and is_active(x) and not is_deleted(x)]

# 풀어 쓰면 디버그하기 좋고 읽기도 좋음
result = []
for x in items:
    if not validate(x):
        continue
    if not is_active(x) or is_deleted(x):
        continue
    result.append(transform(x))