用理解替换Python嵌套的For循环[重复]

Question

这个问题已经在这里有了答案：

• 使用列表理解嵌套循环 3个答案

我正在寻找一种使用列表推导的方法来实现以下目标：

a = [1,2,3,4]
b = [5,6,7,8]
vals = []

for i in a:
  for j in b:
    vals.append(i*j)

print(vals)

我相信有一种方法可以通过列表理解来做到这一点，但是我对如何进行操作感到茫然。

Answer 1

纯列表理解：

[i*j for i in a for j in b]

输出：

[5, 6, 7, 8, 10, 12, 14, 16, 15, 18, 21, 24, 20, 24, 28, 32]

Answer 2

itertools的product将为您提供两个列表中所有元素的组合。 然后，您可以使用一种理解将每对相乘：

from itertools import product
print([x[0] * x[1] for x in product(a, b)])