展平带有因变量的嵌套 for 循环

Question

我要表演：

for i in list_a:
    for j in list_b[i]:
        print(i, j)

是否可以使用 itertools 来完成？我正在寻找类似的东西：

for i, j in itertools.product(list_a, list_b[i])

为了速度和可读性，我想这样做。

Answer 1

itertools 在大多数情况下不会给你速度（但会给你稳定性并节省你的时间所以尽可能使用它）至于可读性和速度 - 没有什么比列表理解更好的了：

your_list = [(i, j) for i in list_a for j in list_b[i]]

然后你可以打印它如果你想:)

Flatten nested for loop with dependent variables