优雅的写双重循环

for i in lit: 
  for j in lit:
    do_something(i, j)

用 itertools 的笛卡尔积

from itertools import product
for i, j in product(lit, lit): 
  do_something(i, j)

product 返回一个生成器,能 yield 出传入数组形成的笛卡尔积。
product 可以接受任意多个数组。

你可能感兴趣的:(优雅的写双重循环)