问题描述
使用 for 循环迭代和连接单词以创建多个排列。我的 for 循环有效,但我不想使用 for 循环,因为我想将代码放在一行上,以便我可以创建一个变量。
我有哪些选择?我意识到的基本问题。
这是我想要连接的两个列表:
first = ["Happy","Sad","Fun"]
second = ["Game","Movie"]
期望输出:
combo = ["Happy Game","Happy Movie","Sad Game","Sad Movie","Fun Game","Fun Movie"]
这个 for 循环有效,但我希望它在一行上并作为一个变量:
for b in first:
for s in second:
combo = b + ' ' + s
print(combo)
有没有办法在没有 for 循环的情况下做到这一点?
解决方法
正如评论中所提到的,列表推导式是在单行中执行此操作的方法:
combos = [b + ' ' + s for b in first for s in second]
这基本上与此相同:
combos = []
for b in first:
for s in second:
combos.append(b + ' ' + s)
结果:
>>> combos
['Happy Game','Happy Movie','Sad Game','Sad Movie','Fun Game','Fun Movie']
有关列表推导式的更多信息,请参见此处:https://docs.python.org/3/tutorial/datastructures.html#list-comprehensions
,您可以使用 itertools.product()
、map()
和 str.join
来实现此结果,而无需使用 for
循环(甚至不在列表推导式中)作为:
>>> from itertools import product
>>> first = ["Happy","Sad","Fun"]
>>> second = ["Game","Movie"]
>>> map(' '.join,product(first,second))
['Happy Game','Fun Movie']
此处itertools.product()
返回可迭代对象的笛卡尔积。
请参阅以下文档了解更多详情: