Mega Code Archive

 

Categories / Python Tutorial / Statement

 

Making a Generator

def flatten(nested):     for sublist in nested:         for element in sublist:             yield element nested = [[1, 2], [3, 4], [5]] for num in flatten(nested):     print num print list(flatten(nested))