Iterator & Generator (yield)

nums = [10, 20, 30]       # list เป็น iterable
it = iter(nums)           # ขอ iterator ออกมา

print(next(it))   # 10
print(next(it))   # 20
print(next(it))   # 30
# next(it) อีกครั้งจะเกิด StopIteration (หมดแล้ว)

# ที่จริง for ทำแบบนี้ให้เราอัตโนมัติ:
for n in nums:
    print(n)

def count_up_to(n):
    i = 1
    while i <= n:
        yield i        # ส่งค่าออก แล้วหยุดพักตรงนี้
        i += 1         # รอบถัดไปมาต่อจากบรรทัดนี้

for num in count_up_to(3):
    print(num)         # 1, 2, 3

# เรียกเฉย ๆ ยังไม่รัน — ได้ generator object
gen = count_up_to(3)
print(gen)             # <generator object ...>
print(next(gen))       # 1

# list comprehension: สร้าง list จริง กิน memory
squares_list = [x * x for x in range(1000000)]

# generator expression: ไม่สร้าง list ผลิตทีละตัว
squares_gen = (x * x for x in range(1000000))

# ใช้กับ sum() ได้เลย ไม่ต้องสร้าง list กลาง
total = sum(x * x for x in range(10))
print(total)   # 285

def read_lines(path):
    with open(path, encoding="utf-8") as f:
        for line in f:          # f เป็น iterator อยู่แล้ว
            yield line.strip()

# วนทีละบรรทัด ไม่ว่าไฟล์ใหญ่แค่ไหนก็ไม่ล้น RAM
# for line in read_lines("huge.txt"):
#     process(line)