二维不定长数组全排列

combination_list.py

import numpy as np
 
 
def search(A):
    nums = len(A)
    max = []
    cur = []
    for i in range(nums):
        max.append(len(A[i]))
        cur.append(0)
    max = np.array(max)
    cur = np.array(cur)
    mul = 1
    for m in max:
        mul = mul*m
 
    result = []
    for _ in range(mul):
        one = np.zeros(nums)
        for i in range(nums):
            print(i)
            one[i] = A[i][cur[i]]
        print(one)
        cur[0] += 1
        for idx in range(nums-1):
            if cur[idx] == max[idx]:
                cur[idx] = 0
                cur[idx+1] += 1
        # print(one)
        result.append(one)
    return result
 
 
if __name__ == '__main__':
    A = [[ '1', '2', '3' ], [ '0', '4' ]]
    print(len(search(A)))