Python中如何不使用for循环修改指定索引对应的列表元素

错误写法说明 #

你之前尝试的列表推导式写法本身不符合Python语法规则：列表推导式内部只能编写表达式，赋值是语句，不能放在推导式中；同时列表推导式的设计目的是生成新列表，不应该用来执行原地修改这类有副作用的操作。

原生Python场景最优实现 #

你目前在用的for循环遍历索引赋值已经是原生Python场景下效率最高的实现，没有额外开销：

my_list = [1, 2, 3, 11, 22, 4, "测试", None]
my_index = [1, 3, 4]

for i in my_index:
    my_list[i] = 1

该方案时间复杂度为O(k)（k为索引列表长度），是该问题的理论下界，同时天然支持保留列表中的混合元素类型。

向量化实现方案（适合大规模数据场景） #

如果数据规模较大（列表长度/索引列表长度超过千级），可以借助NumPy实现向量化批量赋值，不需要在Python层面写循环，执行效率提升明显，同时通过object类型可以保留原列表的混合元素类型：

import numpy as np

my_list = [1, 2, 3, 11, 22, 4, "测试", None]
my_index = [1, 3, 4]

# 转成object类型的NumPy数组，保留所有原始元素类型
arr = np.array(my_list, dtype=object)
# 向量化批量赋值
arr[my_index] = 1
# 转回普通列表
my_list = arr.tolist()

适用场景说明 #

• 小数据量场景下，NumPy方案会有类型转换的额外开销，反而不如原生for循环性能好。
• 如果列表本身都是同构的数值类型，不需要使用object dtype，直接用对应数值类型的NumPy数组即可，性能会比object dtype高3~10倍。

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