最近想用Python写一个简单生成器，类似指定类型和范围，返回指定列表；

 

比如想要  0 ~ 3 的整数，则 我只需要指定：

　　最小：0，

　　最大：3，

　　步长：1

 

则返回一个 [0,1,2,3] 的列表

 

思路是用Python 装饰器，在def方法上打装饰器，装饰器内部替换方法调用时传入的参数值

代码如下：

 

1 # coding=utf-8 2  3 def integer(min=0, max=100, step=1): 4   def deco(func): 5     def wrapper(val): 6       val = [] 7       for i in range(min, max, step): 8         val.append(i)   # 将入参修改掉 9       return func(val)10     return wrapper11   return deco12 13 14 @integer(1, 40)15 def id(val):16   return val17 18 a = id(None)19 print(a)

 

这样 打印 的 a的值就是我要的列表：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22,

23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39]

比Java annotation写起来方便多了，Python大法好！