如何用Python生成符合字母递进规则的JAY前缀字符序列

需求回顾 #

原代码随机生成JAY+数字+随机字符格式的字符串，现在需要改为按字母顺序递进生成，满足：

• 前缀为JAY+数字（数字范围3-6，同原逻辑）
• 后续字母序列按a-z/A-Z顺序递进（支持小写、大写或混合）
• 当字母序列的末尾字符到达z/Z时，触发对应数字递增（如JAY5Abc1d → JAY5Bcd2e）

修改后代码 #

import string

def generate_progressive_sequences(prefix_num=5, char_set=string.ascii_lowercase, letter_length=3, suffix_num_start=1, suffix_letter_start_idx=0):
    # 初始化核心序列状态
    current_letters = list(char_set[:letter_length])
    current_suffix_num = suffix_num_start
    current_suffix_letter = char_set[suffix_letter_start_idx]
    
    while True:
        # 拼接最终序列
        sequence = f"JAY{prefix_num}{''.join(current_letters)}{current_suffix_num}{current_suffix_letter}"
        yield sequence
        
        # 递进后缀字母
        suffix_idx = char_set.index(current_suffix_letter) + 1
        if suffix_idx >= len(char_set):
            # 后缀字母触顶，重置并触发后缀数字递增
            current_suffix_letter = char_set[0]
            current_suffix_num += 1
            
            # 递进主字母序列（从最后一位开始进位）
            idx = len(current_letters) - 1
            carry = True
            while idx >= 0 and carry:
                current_char_idx = char_set.index(current_letters[idx]) + 1
                if current_char_idx >= len(char_set):
                    current_letters[idx] = char_set[0]
                    idx -= 1
                else:
                    current_letters[idx] = char_set[current_char_idx]
                    carry = False
            
            # 主字母全进位，循环递进前缀数字（3-6范围）
            if carry:
                prefix_num = prefix_num + 1 if prefix_num < 6 else 3
                current_letters = list(char_set[:letter_length])
        else:
            current_suffix_letter = char_set[suffix_idx]

# 示例使用：生成混合大小写序列，输出前15个结果
gen = generate_progressive_sequences(prefix_num=5, char_set=string.ascii_letters)
for _ in range(15):
    print(next(gen))

代码关键点说明 #

• 字符集灵活配置：通过char_set参数可切换小写（string.ascii_lowercase）、大写（string.ascii_uppercase）或混合大小写（string.ascii_letters）模式
• 递进逻辑分层：
  1. 优先递进最末尾的后缀字母，触顶后重置并递增后缀数字
  2. 后缀数字递增后，触发主字母序列的进位递进，所有主字母都触顶时，循环切换前缀数字（3→4→5→6→3）
• 可扩展参数：支持调整前缀数字初始值、主字母序列长度、后缀数字起始值等，适配不同格式需求

示例输出（混合字符集） #

JAY5Abc1A
JAY5Abc1B
JAY5Abc1C
...
JAY5Abc1z
JAY5Abc1A
JAY5Abd2A
JAY5Abd2B
...

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