如何高效语法

Question

我写了一些代码。我意识到当我想扩展下面的代码时，它会使语法变得浪费。如何提高效率？

a = ['a', 'b', 'c', 'd', 'e']
b = int(input("Number of digit:\n"))
#This is for 1 digit printed
if b == 1:
   for i in a:
        print(i)
#This is for 2 digit printed
elif b == 2:
    for i in a:
        for j in a:
           print(i,j)

#This is for 3 digit printed
    elif b == 2:
        for i in a:
            for j in a:
                for k in a:
                   print(i,j,k)

Answer 1

itertools.product 会简化您的代码：

import itertools

a = ['a', 'b', 'c', 'd', 'e']
b = int(input("Number of digit:\n"))

for i in itertools.product(a, repeat=b):
    print(" ".join(i))

Answer 2

我认为这可能是要走的路。这将打印您希望元素组合的方式。详细了解 itertools

import itertools
a = ['a','b','c']
num = 2

if num == 2:
  a = [x for x in itertools.product(a, a)]

print(a)  # prints this : [('a', 'a'), ('a', 'b'), ('a', 'c'), ('b', 'a'), ('b', 'b'), ('b', 'c'), ('c', 'a'), ('c', 'b'), ('c', 'c')]

Answer 3

您可以使用 itertools 将此概括为任何 b，如下所示：

import itertools

a = ['a', 'b', 'c', 'd', 'e']
b = int(input("Number of digit:\n"))

for t in itertools.product(a, repeat=b):
    print(*t)

如果您想将元组保存在列表中，您可以这样做：

vals = list(itertools.product(a, repeat=b))

如何高效语法

How to efficiently syntax

python

syntax-highlighting