Python如何根据索引列表修改/提取列表对应元素解决类型报错

Python 原生的 list 类型不支持列表式批量索引赋值，所以会抛出你遇到的类型错误。根据是否可以引入第三方依赖，有两种符合开发规范的高效实现方案：

方案1：仅使用Python标准库（无第三方依赖） #

优先将索引列表转成集合去重，再遍历赋值，避免重复操作，体量越大优化效果越明显：

# 给定输入
a = [2,4,6,8,10,12,14,16,18,20]
b = [2,3,6,9]

# 索引去重，可选添加边界判断避免索引越界：if 0 <= idx < len(a)
for idx in set(b):
    a[idx] = 0

print(a) # 输出结果：[2, 4, 0, 0, 10, 12, 0, 16, 18, 0]

• 优势：无额外依赖，符合Python原生开发规范，时间复杂度为O(n)（n为索引列表b的长度），内存开销极低
• 优化提示：如果b本身已经保证无重复、且所有索引都合法，可以跳过转集合的步骤，执行速度会更快。

方案2：允许使用第三方库（适合大数据量场景） #

如果处理的是数万及以上元素的数值类列表，推荐使用numpy数组，完全支持你提到的a[b] = 0语法，底层由C实现，执行效率远高于原生Python循环：

import numpy as np

a = np.array([2,4,6,8,10,12,14,16,18,20])
b = [2,3,6,9]

a[b] = 0
print(a) # 输出结果：[ 2  4  0  0 10 12  0 16 18  0]

# 如需转回原生Python列表，调用tolist()即可
a = a.tolist()

• 优势：批量操作的速度是原生Python循环的几十到上百倍，非常适合大数据量的数值计算场景。

补充：提取指定索引元素的实现 #

你示例中需要提取对应索引元素生成c的需求，对应实现如下：

• 原生Python实现：

c = [a[idx] for idx in b]

• numpy实现：

c = a[b].tolist()

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