替换 Python 列表中的特定模式索引项



我想替换列表中的每 2 个替代项目。示例如下:

a = [1, 2, 3, 9, 4, 1, 2, 2, 3, 3, 4, 2, 5, 2, 3]

结果应为:

[1, ' ' , ' ', 9,  ' ',  ' ', 2,  ' ', ' ', 3, ' ', ' ', 5, ' ', 3]

最后一项和第一项不应更换

我有循环蛮力枚举方法来做到这一点。我很好奇是否有任何简单和替代的方法可以做到这一点。

就地更改的简单for循环实际上在速度和空间上都是最有效的:

for i in range(1, len(a) - 1):
if i % 3:
a[i] = ' '

这是一个具有列表理解的变体(创建一个新列表(:

lst = [a[0]] + 
[item if n % 3 == 2 else ' ' for n, item in enumerate(a[1:-1])] + 
[a[-1]]

或者如果你想就地做的话更简洁一点:

a[1:-1] = [item if n % 3 == 2 else ' ' for n, item in enumerate(a[1:-1])]

您可以使用切片赋值仅更改需要更改的索引:

a = [1, 2, 3, 9, 4, 1, 2, 2, 3, 3, 4, 2, 5, 2, 3]
a[1:-1:3] = ' ' * len(a[1:-1:3])
a[2:-1:3] = ' ' * len(a[2:-1:3])
# [1, ' ', ' ', 9, ' ', ' ', 2, ' ', ' ', 3, ' ', ' ', 5, ' ', 3]

或者具有更好的空间效率,而不是创建不必要的切片和字符串:

a[1:-1:3] = (' ' for _ in range(1, len(a)-1, 3))
a[2:-1:3] = (' ' for _ in range(2, len(a)-1, 3))

最新更新