python-3.x - 获取 2 个列表的组合
问题描述
我重新发布这个问题是因为我被告知在上一篇文章中有一个解决方案。
我有 2 个列表:
list1 = ["foo", "bar", "lorem"]
list2 = ["X", "Y"]
我想从这两个列表中获得可能的组合,这意味着:
[["foo", "bar", "lorem"],
["foo", "bar", "loremX"],
["foo", "barX", "loremX"],
["fooX", "bar", "loremX"],
["fooX", "barX", "loremX"],
["foo", "barX", "lorem"],
["fooX", "barX", "lorem"],
["fooX", "bar", "lorem"],
["foo", "bar", "lorem"],
["foo", "bar", "loremY"],
["foo", "barY", "loremY"],
["fooY", "bar", "loremY"],
["fooY", "barY", "loremY"],
["foo", "barY", "lorem"],
["fooY", "barY", "lorem"],
["fooY", "bar", "lorem"]]
希望我没有错过任何组合。
有点迷失了这个。
它可能应该是 itertools.combinations_with_replacement
谢谢。
编辑
首先,感谢@titusarmah99 的出色回答。我设法采用了他的第二个非常简单的解决方案并使其通用:
import itertools
list1 = ["foo", "bar", "lorem"]
list2 = ["X", "Y"]
list2new = [""] + list2
newList = [[list1[i]+list2new[j] for j in range(len(list2new))] for i in range(len(list1))]
for index in range(1, len(list2) + 1):
for c in itertools.product([0,index],repeat=len(list1)):
tmp = [newList[i][c[i]] for i in range(len(c))]
print(tmp)
解决方案
这里的关键是使用旋转。有很多方法可以旋转数组,我将使用deque
。
from collections import deque
list1 = ["foo", "bar", "lorem"]
list2 = ["X", "Y"]
list2new = [""] + list2
print(list2new) # ['', 'X', 'Y']
newList = [[list1[i]+list2new[j] for j in range(len(list2new))] for i in range(len(list1))]
print(newList)
# [['foo', 'fooX', 'fooY'], ['bar', 'barX', 'barY'], ['lorem', 'loremX', 'loremY']]
d = deque([0,0,0,1,1,1])
for it in range(2*len(list1)):
tmp = [newList[i][d[i]] for i in range(len(list1))]
print(tmp) #
d.rotate(1)
# ['foo', 'bar', 'lorem']
# ['fooX', 'bar', 'lorem']
# ['fooX', 'barX', 'lorem']
# ['fooX', 'barX', 'loremX']
# ['foo', 'barX', 'loremX']
# ['foo', 'bar', 'loremX']
在此之后,我们可以对 中的其余值重复该过程list2
。
for x in range(1,len(list2new)):
d = deque([0]*len(list1)+[x]*len(list1))
d.rotate(1)
for it in range(2*len(list1)-1): #
tmp = [newList[i][d[i]] for i in range(len(list1))]
print(tmp) #
d.rotate(1)
# ['fooX', 'bar', 'lorem']
# ['fooX', 'barX', 'lorem']
# ['fooX', 'barX', 'loremX']
# ['foo', 'barX', 'loremX']
# ['foo', 'bar', 'loremX']
# ['fooY', 'bar', 'lorem']
# ['fooY', 'barY', 'lorem']
# ['fooY', 'barY', 'loremY']
# ['foo', 'barY', 'loremY']
# ['foo', 'bar', 'loremY']
这将不包含['foo', 'bar', 'lorem']
因此手动添加。
编辑:好像你编辑了这个问题。要获得所有可能的组合,只需使用newList
和itertools.product
for c in itertools.product([0,1],repeat=3):
tmp = [newList[i][c[i]] for i in range(len(c))]
print(tmp) #
推荐阅读
- scala - 从 Akka-2.6.9 更新到 Akka 2.6.10 时 ShardCoordinator$LeastShardAllocationStrategy 中的 NullPointerException
- entity-framework - 为什么这个模型不更新?
- python - 如何在随机游走中使用 imshow 和 colorbar 函数?
- directwrite - DirectWrite 用多种颜色绘制连字?
- node.js - 返回空对象但无法执行 !posts if 语句以在 express 应用程序中使用 mongoose 引发错误
- javascript - 未应用 addEventListener
- mplfinance - 更改 mplfinance 中的支撑线和阻力线颜色
- django - 具有角色和权限的 SSO
- python - 将整数的隐蔽列表转换为字符串
- node.js - 无法在 Npm install 上读取 null 的属性“发射”