python;生成所有元组;多重循环节简化;tuplesGen.py

tuplesGen.py

# -*- coding: utf-8 -*-
"""
 根据给定的范围，用混合进制的方式输出所有元组。
"""

def tuplesGen(ranges):
	"""
	
	>>> ranges = (2,3,4)
	>>> for t in tuplesGen(ranges):
	...     print t
	[0, 0, 0]
	[1, 0, 0]
	[0, 1, 0]
	[1, 1, 0]
	[0, 2, 0]
	[1, 2, 0]
	[0, 0, 1]
	[1, 0, 1]
	[0, 1, 1]
	[1, 1, 1]
	[0, 2, 1]
	[1, 2, 1]
	[0, 0, 2]
	[1, 0, 2]
	[0, 1, 2]
	[1, 1, 2]
	[0, 2, 2]
	[1, 2, 2]
	[0, 0, 3]
	[1, 0, 3]
	[0, 1, 3]
	[1, 1, 3]
	[0, 2, 3]
	[1, 2, 3]
	"""
	n = len(ranges)
	t = [0 for i in xrange(n)]
	while True:
		yield t
		j = 0
		while (j < n and t[j] == ranges[j] - 1):
			t[j] = 0
			j += 1
		if j == n:
			break
		else:
			t[j] += 1

if __name__ == "__main__":
	import doctest
	doctest.testmod()