在 Python 中用新值替换替换索引中给定位置的列表中的元素

Question

我想用新值替换替换索引中给定位置的列表中的元素。例如，如果给定的列表是，replace_elements([1,2,3,4,5,6,7,8],[0,4,3],0)那么结果应该是 [0, 2, 3, 0, 0, 6, 7, 8]。我尝试了几种方法，但它们似乎都不起作用。我尝试了以下代码：for i in new_value:   list_a[i] = replacement_indicesreturn new_value

Answer 1

一种不能就地工作的列表理解方法：

只需根据列表理解中的索引在三元表达式中决定替换值或原始值：

def replace_elements(inlist, indexes, replvalue):

    return [replvalue if i in indexes else x for i,x in enumerate(inlist)]

print(replace_elements([1,2,3,4,5,6,7,8],[0,4,3],0))

结果：

[0, 2, 3, 0, 0, 6, 7, 8]

对于大索引列表，[0,4,3]应该是set( {0,4,3}) 以加快查找速度。

Answer 2

你把replacment_indices和new_value放在错误的地方。您应该在每个指定的索引处迭代replacement_indices并分配new_value给列表：

for i in replacement_indices:
   list_a[i] = new_value

由于您正在就地修改列表，因此也无需返回任何内容，这意味着循环之后list_a将根据规范进行修改。