デザインパターンの視点から見たPythonのジェネレータ

def total(iterable):
    total = 0
    for v in iterable:
        total += v
    return total

iterable = [1, 10, 100]
total(iterable) # => 111

def generator1():
    yield 1
    yield 10
    yield 100

iterable = generator1()
total(iterable) # => 111

def generator2(n):
    for i in range(n):
        yield 10

iterable = generator2(5)
total(iterable) # => 50

def generator3(word_list):
    for word in word_list:
        yield len(word)

iterable = generator3(['hoge', 'fuga', 'piyo'])
total(iterable) # => 12

sum(iterable[, start])

a,b,c
d,e,f

g,h,i
j,k,l

import csv

# 1つのCSVファイルを読み込んで全ての行を1つのリストに結合する関数
def concat_csv_rows(file_path):
    result = []
    with open(file_path) as f:
        reader = csv.reader(f)
        for row in reader:
            result.extend(row) # 各行の結合
    return result

print(concat_csv_rows('file1.csv'))
# =>
# ['a', 'b', 'c', 'd', 'e', 'f']

# 複数のファイルをリストで受け取れるように拡張した関数
def concat_csv_rows_multi(file_path_list):
    result = []
    for file_path in file_path_list:
        result.extend(concat_csv_rows(file_path)) # 各ファイルの処理結果を結合
    return result

print(concat_csv_rows_multi(['file1.csv', 'file2.csv']))
# =>
# ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l']

# 1行ずつ行を返すイテレータを受け取って結合する関数
def concat_rows(iter_rows):
    result = []
    for row in iter_rows:
        result.extend(row) # 各行の結合
    return result

import csv

# 1つのファイルを読み込んで1行ずつイテレートするジェネレータ
def get_csv_rows(file_path):
    with open(file_path) as f:
        reader = csv.reader(f)
        for row in reader:
            yield row

print(concat_rows(get_csv_rows('file1.csv')))
# =>
# ['a', 'b', 'c', 'd', 'e', 'f']

# 複数のファイルをリストで受け取れるように拡張したジェネレータ
def get_csv_rows_multi(file_path_list):
    for file_path in file_path_list:
        yield from get_csv_rows(file_path) # ジェネレータをループで呼び出すのと等価

print(concat_rows(get_csv_rows_multi(['file1.csv', 'file2.csv'])))
# =>
# ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l']

def get_csv_rows_multi(file_path_list):
    for file_path in file_path_list:
        for row in get_csv_rows(file_path):
            yield row

rows = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
print(concat_rows(rows))
# =>
# [1, 2, 3, 4, 5, 6, 7, 8, 9]