算法：用itertools.product()简化嵌套for循环

今天这一题叫做“偷瞄到的密码”：

警察跟踪一名窃贼来到了一个仓库门前。仓库的密码锁盘如下：

1 2 3

4 5 6

7 8 9

   0

窃贼输入密码后进了门。警察“觉得”自己看到了密码比如1357，但是也有可能是相邻的数字（相邻仅包括正上下左右，不包括对角线），比如第一位不是1，而是相邻的4和2（不包括5）。

可能的密码是哪些组合呢？

这道题目可以抽象成：

1. 0-9各自都对应了一组相邻数字

现给定一个数字串：

2. 对数字串中的每个数字，找到对应的相邻数组

3. 计算从这些相邻组中各挑一个形成的所有可能组合并输出

实现要点：

1. 上述#3实际上是一个嵌套的多重for循环：

　　for 数字串中的每个数字:

　　　　for 该数字相邻数组中的每个数字:

用itertools模块的product方法可以将这个过程简化为一步完成。根据Python手册，product(A, B)的结果是((x,y) for x in A for y in B)。

2. 对于函数或者方法如product(A, B, C, D)，如果ABCD在一个列表里ls=[A, B, C, D]，如何把ls传给product？

用*，即product(*ls)。

3. product的输出为元组。用join合并元组的所有元素，以字符串的形式输出密码组合。

Python实现：

#-*- coding:utf-8 -*-
__author__ = 'Administrator'

from itertools import product
import time

combine=["08","124","1235","236","1457","24568","3569","478",
         "57890","689"]

def get_pin(observe):
    return ["".join(p) for p in product(*(combine[int(d)] for d in observe))]

def get_pin2(observe):
    comb=[]
    li=[]
    for d in observe:
        comb.append(combine[int(d)])
    for p in product(*comb):
        li.append("".join(p))
    return li

if __name__=="__main__":
    t0=time.time()
    for i in range(10000):
        get_pin("1357")
    t1=time.time()-t0
    print(t1)
    print("-"*100)
    t0=time.time()
    for i in range(10000):
        get_pin2("1357")
    t1=time.time()-t0
    print(t1)

输出时间对比：

0.6550376415252686
----------------------------------------------------------------------------------------------------
0.7860448360443115