使用因变量展平嵌套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]]

然后，如果您愿意，可以打印它：）