JavaScript双层索引赋值语法的Python等价实现方法

原始参考JS代码 #

这是一段遗传算法排列交叉的实现片段，原始代码如下：

// applying crossover
for (let i = cxpoint1; i < cxpoint2; i++) {
  let temp1 = ind1[i]
  let temp2 = ind2[i]

  ind1[i]         = temp2
  ind1[p1[temp2]] = temp1

  ind2[i]         = temp1
  ind2[p2[temp1]] = temp2

  return [ind1, ind2]
}

注意：原代码把return写在for循环内部属于笔误，执行第一次循环就会直接退出，正常逻辑需要遍历完交叉区间所有位置后再返回结果。

目标语句拆解 #

你问的ind1[p1[temp2]] = temp1是非常常规的多层索引写法，Python原生完全支持这种语法，不需要做特殊转换，执行顺序是从内向外逐层计算：

• 第一步计算内层索引：取当前变量temp2的值，作为键去映射表p1中查询对应的位置，得到要赋值的目标下标
• 第二步执行赋值操作：把ind1数组对应目标下标的元素值，设置为temp1

拆成等价的分步写法逻辑完全一致，不管JS还是Python都通用：

# 拆分后的等价写法
target_index = p1[temp2]
ind1[target_index] = temp1

业务逻辑背景 #

这段是部分匹配交叉（PMX，排列编码问题最常用的交叉算子）的核心冲突修复逻辑：

• ind1、ind2是两个待交叉的个体，为排列格式（比如TSP问题的城市访问顺序，所有元素唯一不重复）
• p1、p2是两个个体的值-位置映射表，p1[val]存储的是值val在ind1中对应的下标位置
• 当你把ind1在i位置的原值temp1替换成ind2的temp2后，ind1里就会出现两个temp2（一个在刚替换的i位置，一个在原来的p1[temp2]位置），这行代码的作用就是把原来存temp2的位置替换成temp1，消除重复值，保证排列合法性。

Python等价实现 #

和原JS逻辑完全对齐的Python实现如下，你可以直接适配到自己的代码里：

def pmx_crossover(ind1: list, ind2: list, cxpoint1: int, cxpoint2: int):
    # 构建值到位置的映射表，对应JS代码中的p1、p2
    pos1 = {val: idx for idx, val in enumerate(ind1)}
    pos2 = {val: idx for idx, val in enumerate(ind2)}

    for i in range(cxpoint1, cxpoint2):
        val1 = ind1[i]
        val2 = ind2[i]

        # 交换当前交叉位的值
        ind1[i] = val2
        # 就是你问的双层索引语句，Python直接支持，写法和JS完全一致
        ind1[pos1[val2]] = val1
        # 同步更新映射表
        pos1[val1], pos1[val2] = pos1[val2], pos1[val1]

        ind2[i] = val1
        ind2[pos2[val1]] = val2
        pos2[val1], pos2[val2] = pos2[val2], pos2[val1]

    return [ind1, ind2]

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