如何根据输入（行数）找到可能字符串的概率

Question

如何根据输入（行数）找到可能字符串的概率。你知道吗

假设我们有如下9x3级的矩阵，即9行3列。你知道吗

A B C
D E F
G H I
. . .
. . .
X Y Z

用户可以提供1到9之间的任意数字 所以，如果用户提供2（行数） 输出应该如下所示

AD
AE
AF
BD
BE
BF
CD
CE
CF

如果用户提供输入3，则输出应该如下所示

ADG
ADH
ADI
AEG
AEH
AEI
AFG
AFH
AFI
BDG
BDH
BDI
BEG
BEH
BEI
BFG
BFH
BFI
CDG
CDH
CDI
CEG
CEH
CEI
CFG
CFH
CFI

类似地，输入行数可以从1到9不等 如何在python中实现这一点？有没有最好的办法？ 最近，我在一次采访中遇到了这个问题。我解决不了这个问题。 也许这篇文章会对很多人有所帮助。你知道吗

Answer 1

您可以利用itertools模块中的product()函数。你知道吗

import itertools

for i in itertools.product(*A[:N]):
    print(''.join(i))

假设A是您的矩阵，N是要选择的行数。你知道吗

---

这里有一个简短的解释。要使用N=2生成列表，您将使用两个for循环：

for r1 in A[0]:
    for r2 in A[1]:
        print(''.join([r1, r2]))

对于N=3，您将使用三个For循环：

for r1 in A[0]:
    for r2 in A[1]:
        for r3 in A[2]:
            print(''.join([r1, r2, r3]))

因此，一般来说，需要N嵌套for循环，这就是product()生成的内容。你知道吗