python yield

描述：遍历数据，数据为数字或者由数字填充的列表

不使用生成器

data = [[1, 2], [3, 2], [5]]

for lst in data:
    for item in lst:
        print item

使用生成器

data = [[1, 2], [3, 2], [5]]

def flatten(nested):
    for lst in nested:
        for element in lst:
            yield element

for item in flatten(data):
    print item

这里是直接打印，但是如果需要得到一个一维的列表作为结果，不使用生成器需要一个在循环内随时可以写的 [] 来每次append，而使用生成器则直接list(flatten(data))

 

如果嵌套的层数不定

使用生成器

data_1 = [1, 3]
data_2 = [[3, 1], [4, 1]]
data_3 = [1, 2]

data = [data_1, data_2, data_3]

def flatten(nested):
    try:
        for lst in nested:
            for item in flatten(lst):
                yield item
    except:
        yield nested

for item in flatten(data):
   print item

 不使用生成器

data_1 = [1, 3]
data_2 = [[3, 1], [4, 1]]
data_3 = [1, 2]

data = [data_1, data_2, data_3]

def foo(data):
    for item in data:
        if type(item) is type(1):
            print item,
        else:
            foo(item)

foo(data)

参考：《python基础教程》 人民邮电出版社