求助：用Python生成可通过索引访问的字母组合数组

需要生成一个按单字母→双字母→多字母顺序排列的小写字母组合数组，要求数组索引从1开始对应第一个元素（如array[1]='a'、array[2]='b'），后续依次为'c'...'z'、'aa'、'ab'...'zz'、'aaa'等。

解决方案代码 #

from string import ascii_lowercase
import itertools

def generate_string_array(element_count):
    # 生成器：按长度递增生成所有小写字母组合
    def iter_all_strings():
        for size in itertools.count(1):
            for s in itertools.product(ascii_lowercase, repeat=size):
                yield "".join(s)
    
    # 初始化数组，用空字符串占位让索引从1开始对应目标元素
    result_array = ['']
    # 截取指定数量的元素并添加到数组
    for s in itertools.islice(iter_all_strings(), element_count):
        result_array.append(s)
    
    return result_array

# 示例：生成包含前1000个元素的数组
target_array = generate_string_array(1000)
# 验证索引访问
print(target_array[1])   # 输出: 'a'
print(target_array[26])  # 输出: 'z'
print(target_array[27])  # 输出: 'aa'

代码说明 #

• iter_all_strings生成器通过itertools.count不断递增字符串长度，结合itertools.product生成对应长度的所有字母组合
• generate_string_array函数接收需要生成的元素数量，返回的数组在开头添加空字符串占位，确保索引1对应第一个字母组合'a'
• 使用itertools.islice控制生成的元素数量，避免无限生成

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