简单的列表表达式:

sq = [ x**2 for x in range(4)]
for s in sq:
    print s 

结果:
0
1
4
9