Python列表元素插入问题：按指定索引插入0的代码修正

我有一个列表x0，想要根据missing_values里指定的位置插入0。以下是现有代码、当前输出和期望输出：

现有代码 #

x0 = [1, 2, 3, 4, 5]
missing_values = [2, 4]  # 要插入0的索引位置

# 将需要处理的索引转为集合
indices_to_set_zero = set(missing_values)

# 初始化结果列表
result = []

# 遍历原列表
for i, value in enumerate(x0):
    # 如果当前索引在集合中，先追加0再追加当前元素
    if i in indices_to_set_zero:
        result.append(0)
    result.append(value)

# 如果原列表长度在集合中，在末尾追加0
if len(x0) in indices_to_set_zero:
    result.append(0)

print(result)

当前输出 #

[1, 2, 0, 3, 4, 0, 5]

期望输出 #

[1, 0, 2, 0, 3, 4, 5]

问题分析与修正 #

原代码的逻辑是在原列表索引匹配missing_values的元素前插入0，这导致0出现在原元素3、5的前面，和期望输出不符。

期望输出的0要插在原元素1、2之间，以及原元素2、3之间，对应原列表的索引1、2的元素前插入0。

方案一：修正missing_values匹配原逻辑 #

如果你的需求是“在原列表指定索引的元素前插入0”，只需将missing_values改为[1, 2]，原代码即可输出期望结果：

x0 = [1, 2, 3, 4, 5]
missing_values = [1, 2]  # 修改为目标索引
indices_to_set_zero = set(missing_values)

result = []
for i, value in enumerate(x0):
    if i in indices_to_set_zero:
        result.append(0)
    result.append(value)

if len(x0) in indices_to_set_zero:
    result.append(0)

print(result)

输出：

[1, 0, 2, 0, 3, 4, 5]

方案二：按新列表位置插入0 #

如果missing_values是指新列表中要设置为0的位置（比如期望输出中0在索引1、3），可以用以下逻辑实现：

x0 = [1, 2, 3, 4, 5]
missing_values = [1, 3]  # 新列表中0的位置
total_length = len(x0) + len(missing_values)
# 计算原元素在新列表中的对应位置
original_positions = sorted(set(range(total_length)) - set(missing_values))

result = [0] * total_length
for idx, val in enumerate(x0):
    result[original_positions[idx]] = val

print(result)

输出同样为期望的[1, 0, 2, 0, 3, 4, 5]。

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