|
- def flatten(nested):
- try:
- for sublist in nested:
- for element in flatten(sublist):
- print 'nested'
- yield element
- except TypeError:
- print "------"
- print 'flatten'
- yield nested
- a = [[[1],2],3]
- b=flatten(a)
- while True:
- print '*********'
- print b.next()
复制代码
output:
- *********
- ------
- flatten
- nested
- nested
- nested
- 1
- *********
- ------
- flatten
- nested
- nested
- 2
- *********
- ------
- flatten
- nested
- 3
- *********
复制代码 |
|