Python 按owner列表长度批量关联gizmos元素实现双列表组合

直接使用zip函数无法实现需求的原因是：zip会以最短输入列表的长度为边界，做一对一配对，即owner1配对第一个gizmo元素、owner3配对第二个gizmo元素、owner32配对第三个gizmo元素，直接截断剩余的gizmos元素，完全不符合“以owner长度为固定步长拆分gizmos分组绑定”的规则。

实现逻辑 #

按照需求明确的匹配规则拆解步骤：

• 取固定步长为len(owner)（本案例为3）
• 前len(owner)-1个分组，每组严格取步长个连续gizmos元素，按顺序绑定对应位置的owner
• 最后一个owner绑定gizmos列表中所有剩余的元素，不要求剩余元素长度严格等于步长
• 对每个绑定关系，将owner字符串插入组内原元组的首位，拼接为新元组

可落地代码实现 #

step = len(owner)
new_gizmos = []

# 处理前两个完整步长分组，绑定对应位置的owner
for idx in range(len(owner) - 1):
    current_owner = owner[idx]
    start = idx * step
    end = start + step
    current_group = gizmos[start:end]
    # 为组内每个元组前置当前owner，批量加入新列表
    new_gizmos.extend( (current_owner, *item) for item in current_group )

# 处理最后一个owner，绑定所有剩余的gizmos元素
last_owner = owner[-1]
last_start = (len(owner) - 1) * step
last_group = gizmos[last_start:]
new_gizmos.extend( (last_owner, *item) for item in last_group )

运行验证 #

代码生成的前9个元素，和需求的目标效果完全匹配：

new_gizmos[0:9]
# 输出结果：
# [('owner1', 'gizmo1', 314),
#  ('owner1', 'gizmo1', 18),
#  ('owner1', 'gizmo1', 72),
#  ('owner3', 'gizmo1', 2),
#  ('owner3', 'gizmo1', 252),
#  ('owner3', 'gizmo1', 314),
#  ('owner32', 'gizmo1', 18),
#  ('owner32', 'gizmo1', 72),
#  ('owner32', 'gizmo1', 2)]

适配说明 #

• 代码没有硬编码步长数值，后续如果owner列表长度变化，无需手动修改步长值，可自动适配
• 元组拼接使用解包语法*item，不需要硬编码原元组的内部元素个数，哪怕gizmos的元组是任意长度都能正常拼接
• Python列表切片天然支持越界处理，哪怕gizmos总长度小于步长，也不会触发索引报错

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