gpt4 book ai didi

python - Itertools 排列

转载 作者:行者123 更新时间:2023-11-28 22:23:15 26 4
gpt4 key购买 nike

我有一个列表 x=[1,2,3,4,5] 并且想查看这个列表的不同排列,一次取两个数字。

x=[1,2,3,4,5] 
from itertools import permutations
y=list(i for i in permutations(x,2) if i[0]<i[1])
print(y)

输出:[(1, 2), (1, 3), (1, 4), (1, 5), (2, 3), (2, 4), (2, 5) , (3, 4), (3, 5), (4, 5)]

但我还想在输出中使用 [(1,1),(2,2),(3,3),(4,4),(5,5)]。如何纠正这个?

最佳答案

你想要combinations_with_replacement()相反:

>>> from itertools import combinations_with_replacement
>>> list(combinations_with_replacement(x, 2))
[(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (2, 2), (2, 3), (2, 4), (2, 5), (3, 3), (3, 4), (3, 5), (4, 4), (4, 5), (5, 5)]

关于python - Itertools 排列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47122201/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com