生成范围内的连续字符字符串

Question

我想生成从 "aaa" 到 "zzz" 的所有字符串。目前，我是用三个循环来实现这个功能，有没有更简洁、更符合 Python 风格的方法呢？

key_options = []
for n1 in range(ord('a'), ord('z')+1):
    for n2 in range(ord('a'), ord('z')+1):
        for n3 in range(ord('a'), ord('z')+1):
             key_options.append(chr(n1) + chr(n2) + chr(n3))

Answer 1

>>> letters = [chr(i) for i in range(ord('a'), ord('z')+1)]
>>> letters
['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']

import itertools
["".join(i) for i in itertools.product(letters,letters,letters)]

输出

['aaa', 'aab', 'aac', ... 'zzy', 'zzz']

Answer 2

itertools模块是处理这种循环的更好方法。你可以使用它的product函数：

itertools.product(*iterables[, repeat])

这是输入可迭代对象的笛卡尔积。

这相当于在生成器表达式中使用嵌套的for循环。例如，product(A, B)的结果和((x,y) for x in A for y in B)是一样的。

string模块可以提供ASCII小写字母，而不需要使用range：

string.ascii_lowercase

小写字母为'abcdefghijklmnopqrstuvwxyz'。这个值不依赖于地区设置，始终保持不变。

因此，你可以得到

from itertools import product
from string import string

key_options = [''.join(n) for n in product(ascii_lowercase, repeat=3)]

Answer 3

使用 itertools.product 和列表推导式：

>>> from itertools import product
>>> from string import ascii_lowercase
>>> [''.join(p) for p in product(ascii_lowercase, repeat=3)]
['aaa', 'aab', 'aac', 'aad', 'aae', ..., 'zzv', 'zzw', 'zzx', 'zzy', 'zzz']