优雅的写双重循环

2016-08-11  本文已影响36人  知识分子中的文盲
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 可以接受任意多个数组。

上一篇下一篇

猜你喜欢

热点阅读