Python如何修改字母组合生成代码，使其从指定自定义字符串开始生成

实现思路 #

将小写字母序列对应为26进制的计数逻辑，无需遍历起始点之前的所有组合，直接从指定字符串开始生成后续序列：

1. 先将输入的起始字符串转换为每个字符对应ascii_lowercase的索引元组
2. 先通过进位逻辑生成与起始字符串长度相同的所有后续组合
3. 长度相同的组合生成完毕后，再依次生成更长长度的所有字母组合

完整代码 #

from string import ascii_lowercase
import itertools

def increment_indices(indices):
    # 处理索引元组的进位逻辑，从最右侧位开始加1
    i = len(indices) - 1
    while i >= 0:
        indices[i] += 1
        if indices[i] < 26:
            return indices
        indices[i] = 0
        i -= 1
    # 所有位均进位，说明需要切换到更长的长度
    return None

def iter_strings_from(start: str):
    # 校验输入仅包含小写字母
    if not all(c in ascii_lowercase for c in start):
        raise ValueError("起始字符串仅支持小写英文字母")
    
    char_to_idx = {c: i for i, c in enumerate(ascii_lowercase)}
    start_indices = tuple(char_to_idx[c] for c in start)
    start_len = len(start_indices)

    # 生成与起始字符串长度相同的后续组合
    current = list(start_indices)
    while current is not None:
        yield ''.join(ascii_lowercase[i] for i in current)
        current = increment_indices(current)
    
    # 生成更长长度的所有组合
    for size in itertools.count(start_len + 1):
        for s in itertools.product(ascii_lowercase, repeat=size):
            yield ''.join(s)

使用示例 #

如果需要生成从bbbb到cccc的序列，使用方式如下：

for s in iter_strings_from('bbbb'):
    print(s)
    if s == 'cccc':
        break

运行后会直接从bbbb开始输出，不会遍历前面的a到aaaa等内容，中断重启后不需要重新生成历史序列，效率大幅提升。

内容的提问来源于stack exchange，提问作者Sven