如何仅对列表L中指定索引I的元素应用映射函数g？

这个需求我之前也碰到过，不用写复杂的循环判断，有好几种简洁的方式能搞定。结合你给出的例子（L = [1,2,3,4,5,6,7]，索引集合I = {1,3,4}，平方函数g(x)=x²），我给你整理几种常用方法：

方法1：列表推导式（Python首选） #

这是Python里最清爽的写法，用enumerate同时拿到索引和元素，直接判断处理：

L = [1, 2, 3, 4, 5, 6, 7]
target_indices = {1, 3, 4}  # 用集合是为了更快的查找
square = lambda x: x ** 2

result = [square(num) if idx in target_indices else num for idx, num in enumerate(L)]
print(result)  # 输出: [1, 4, 3, 16, 25, 6, 7]

这里把索引列表转成集合很关键——集合的查找是O(1)时间复杂度，比用列表查找快得多，尤其是当目标索引数量很多的时候。

方法2：原地修改（省内存的选择） #

如果不想生成新列表，直接修改原列表的话，代码更简单：

for idx in target_indices:
    L[idx] = square(L[idx])

这种方式没有额外的内存开销，适合处理大型列表，直接定位到目标索引修改就行，逻辑一目了然。

其他语言的类似实现（以JavaScript为例） #

核心思路其实是通用的，换个语言也一样清爽：

const L = [1,2,3,4,5,6,7];
const targetIndices = new Set([1,3,4]);
const square = x => x ** 2;

const result = L.map((num, idx) => targetIndices.has(idx) ? square(num) : num);
console.log(result); // 输出: [1,4,3,16,25,6,7]

小提示 #

• 不管用哪种语言，尽量把目标索引转换成集合类型，避免线性查找带来的性能损耗。
• 如果你的索引列表里可能有重复值，建议先去重（比如Python里用set(I)），避免重复修改同一个元素。

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