Python中一组中3个数字的所有可能组合

时间:2014-03-30 20:32:35

标签: python algorithm combinatorics

我想打印套装中的3个数字的所有可能组合( 0 ... n-1 ),而这些组合中的每一个都是唯一的。我通过以下代码获得变量 n

n = raw_input("Please enter n: ")

但我仍然坚持提出算法。有什么帮助吗?

4 个答案:

答案 0 :(得分:10)

from itertools import combinations
list(combinations(range(n),3))

只要您使用的时间晚于Python 2.6

,这将有效

答案 1 :(得分:2)

itertools是您的朋友,特别是permutations

演示:

from itertools import permutations 

for item in permutations(range(n), 3): 
    print item

假设你有Python 2.6或更新版本。

答案 2 :(得分:1)

如果您希望所有可能的组合重复值并且位置不同,则需要使用以下产品:

from itertools import product
t = range(n)
print set(product(set(t),repeat = 3))

例如,如果n = 3,则输出为:

set([(0, 1, 1), (1, 1, 0), (1, 0, 0), (0, 0, 1), (1, 0, 1), (0, 0, 0), (0, 1, 0), (1, 1, 1)])

希望这会有所帮助

答案 3 :(得分:0)

combos = []
for x in xrange(n):
    for y in xrange(n):
        for z in xrange(n):
             combos.append([x,y,z])