多个数组，每个数组中选择一项，总共能有多少种组合，reduce函数实现

问题来源：《有意思的面试题-数独问题》
我在《多个数组，每个数组中选择一项，总共能有多少种组合，treelib实现》中写了通过treelib实现的方式，但我发现这个方式内存消耗量极大，有的时候内存使用量达到30G，reduce对内存的使用也很大，但性能上胜出，所以这次使用reduce搞一把。
废话不多说了，看代码：

import numpy as np
import sys

def combination(lists):

    if sys.version_info < (3, 0):
        import reduce
    else:
        from functools import reduce

    def reduceFun(list1, list2):
        array = []
        for i in list1:
            for j in list2:
                array.append(np.hstack((i, j)))
        return array

    return reduce(reduceFun, lists)

#lists =  [[6, 8], [2, 5], [1, 9], [3, 9], [3, 5], [3, 9], [8, 9], [3, 6], [2, 9], [2, 3, 8], [2, 3, 6], [2, 5, 8], [1, 5, 8], [1, 6, 9], [5, 6, 9], [1, 3, 5], [3, 5, 6], [3, 8, 9]]
lists =  [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

combin = combination(lists)
print(len(combin))
print(combin)

测试结果：

27
[array([1, 4, 7]), array([1, 4, 8]), array([1, 4, 9]), array([1, 5, 7]), array([1, 5, 8]), array([1, 5, 9]), array([1, 6, 7]), array([1, 6, 8]), array([1, 6, 9]), array([2, 4, 7]), array([2, 4, 8]), array([2, 4, 9]), array([2, 5, 7]), array([2, 5, 8]), array([2, 5, 9]), array([2, 6, 7]), array([2, 6, 8]), array([2, 6, 9]), array([3, 4, 7]), array([3, 4, 8]), array([3, 4, 9]), array([3, 5, 7]), array([3, 5, 8]), array([3, 5, 9]), array([3, 6, 7]), array([3, 6, 8]), array([3, 6, 9])]

如果您感兴趣，就将上面的那串长的注释打开，效果也很不错呢。