我知道itertools,但它似乎只能生成没有重复的排列。例如,我想为2个骰子生成所有可能的掷骰,所以我需要[1,2,3,4,5,6]的大小为2的所有排列,包括重复:(1,1)、(1,2)、(2,1)......等等如果可能,我不想从头开始实现
2cmtqfgy1#
您正在寻找Cartesian Product。在数学中,笛卡尔积(或乘积集)是两个集合的直积。在您的情况下,这将是{1, 2, 3, 4, 5, 6} x {1, 2, 3, 4, 5, 6}。itertools可以帮助您:
{1, 2, 3, 4, 5, 6}
itertools
import itertools x = [1, 2, 3, 4, 5, 6] [p for p in itertools.product(x, repeat=2)] [(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (2, 6), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (3, 6), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (4, 6), (5, 1), (5, 2), (5, 3), (5, 4), (5, 5), (5, 6), (6, 1), (6, 2), (6, 3), (6, 4), (6, 5), (6, 6)]
随机掷骰子(效率极低):
import random random.choice([p for p in itertools.product(x, repeat=2)]) (6, 3)
31moq8wy2#
你不是在寻找排列--你需要的是Cartesian Product。
from itertools import product for roll in product([1, 2, 3, 4, 5, 6], repeat = 2): print(roll)
vzgqcmou3#
在Python 2.7和Python 3.1中有一个itertools.combinations_with_replacement函数:
itertools.combinations_with_replacement
>>> list(itertools.combinations_with_replacement([1, 2, 3, 4, 5, 6], 2)) [(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (2, 2), (2, 3), (2, 4), (2, 5), (2, 6), (3, 3), (3, 4), (3, 5), (3, 6), (4, 4), (4, 5), (4, 6), (5, 5), (5, 6), (6, 6)]
igsr9ssn4#
在这种情况下,并不特别需要列表解析。
给定
import itertools as it seq = range(1, 7) r = 2
代码
list(it.product(seq, repeat=r))
详细数据
显然,笛卡尔积可以生成排列的子集,但它遵循:
product
[x for x in it.product(seq, repeat=r)]
第一个因此,所有组合函数都可以从product实现:
combinations_with_replacement
permutations
combinations
t3irkdon5#
我想我找到了仅使用lambdas、map和reduce的解决方案。
lambdas
map
reduce
product_function = lambda n: reduce(lambda x, y: x+y, map(lambda i: list(map(lambda j: (i, j), np.arange(n))), np.arange(n)), [])
本质上我是在Map第一个lambda函数,它给定一行,迭代列
list(map(lambda j: (i, j), np.arange(n)))
则将其用作新lambda函数的输出
lambda i:list(map(lambda j: (i, j), np.arange(n)))
Map到所有可能的行
map(lambda i: list(map(lambda j: (i, j), np.arange(n))), np.arange(m))
然后将所有结果列表缩减为一个列表。
也可以使用两个不同的数字。
prod= lambda n, m: reduce(lambda x, y: x+y, map(lambda i: list(map(lambda j: (i, j), np.arange(m))), np.arange(n)), [])
l7mqbcuq6#
首先,你需要把itertools返回的生成器.permutations(list)转换成一个列表.然后,你可以使用set()来删除重复项如下所示:
def permutate(a_list): import itertools return set(list(itertools.permutations(a_list)))
6条答案
按热度按时间2cmtqfgy1#
您正在寻找Cartesian Product。
在数学中,笛卡尔积(或乘积集)是两个集合的直积。
在您的情况下,这将是
{1, 2, 3, 4, 5, 6}
x{1, 2, 3, 4, 5, 6}
。itertools
可以帮助您:随机掷骰子(效率极低):
31moq8wy2#
你不是在寻找排列--你需要的是Cartesian Product。
vzgqcmou3#
在Python 2.7和Python 3.1中有一个
itertools.combinations_with_replacement
函数:igsr9ssn4#
在这种情况下,并不特别需要列表解析。
给定
代码
详细数据
显然,笛卡尔积可以生成排列的子集,但它遵循:
product
产生所有排列nr第一个
因此,所有组合函数都可以从
product
实现:product
实作的combinations_with_replacement
permutations
实现combinations
,permutations
可以用product
实现(见上文)t3irkdon5#
我想我找到了仅使用
lambdas
、map
和reduce
的解决方案。本质上我是在Map第一个lambda函数,它给定一行,迭代列
则将其用作新lambda函数的输出
Map到所有可能的行
然后将所有结果列表缩减为一个列表。
甚至更好
也可以使用两个不同的数字。
l7mqbcuq6#
首先,你需要把itertools返回的生成器.permutations(list)转换成一个列表.然后,你可以使用set()来删除重复项如下所示: