Python循环拼接字符串生成指定序列列表报错修复方法

问题目标 #

• 当参数m=3时，生成交替列表['a1', 'b1', 'a2', 'b2','a3', 'b3']，要求逻辑支持m替换为任意更大数值（比如100）
• 可扩展支持多前缀场景，比如前缀为a/b/c时生成['a1', 'b1', 'c1', 'a2', 'b2','c2','a3', 'b3', 'c3']

原有代码报错原因 #

两次尝试报错本质都是类型/运算逻辑不匹配：

1. 纯列表写法[ ['a','b'] + str(i) for i in np.arange(1,m+1)]触发TypeError: can only concatenate list (not "str") to list：Python中列表的+运算符仅支持拼接同类型列表，无法直接和字符串做拼接运算。
2. numpy写法触发UFuncTypeError：直接将一维字符数组和单个字符串做加法时，numpy无法匹配到对应的广播运算规则，字符串拼接的通用函数无法执行。

最简实现（纯Python无依赖） #

不需要引入numpy，用两层循环的列表推导即可实现，修改前缀列表就能支持任意数量的前缀：

m = 3
# 要扩展前缀只需要修改这个列表即可，比如改成['a','b','c']就支持三前缀
prefix_list = ['a', 'b']

result = [prefix + str(num) for num in range(1, m+1) for prefix in prefix_list]

运行上述代码，当prefix_list = ['a','b']时输出为['a1', 'b1', 'a2', 'b2', 'a3', 'b3']；当prefix_list = ['a','b','c']时输出为['a1', 'b1', 'c1', 'a2', 'b2', 'c2', 'a3', 'b3', 'c3']，m修改为100等任意正整数都可以直接运行。

大规模数据下的numpy优化实现 #

如果m取值非常大（比如超过10万），可以用numpy的广播机制提升运算效率：

import numpy as np
m = 3
prefix_list = np.array(['a', 'b', 'c'])

# 按列优先展平二维数组，得到和纯Python写法完全一致的顺序
result = (prefix_list[:, None] + np.arange(1, m+1).astype(str)).flatten(order='F').tolist()

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