从列表中选择第n个值，然后在它们之间替换值

Question

我在python中有一个列表，其中包含一些值：List = [0,1,2,3,4,5,6,7,8,9,1,2,3,4,5,6,7,8， 9，...，n]我想做的是选择第n个值，并用零填充它们之间的距离。 这样我最终得到以下列表：结果= [0,0,0,0,0,5,0,0,0,0,0,0,0,0,5,0,0,0,0，.. 。，n]注意：并不总是5。

我已完成的工作：编写类似于窗口函数的内容并遍历列表。

def window(x):
    window = []
    for ii in range(len(x)):
        mask = [0,1,2,3,None,5,6,7,8]
        select = [x is None for x in mask]
        center = list(itertools.compress(x[ii],select))
        window.append(center)
        for ii in range(0,4):
            center.insert(0,"0")
            center.append("0")
    return window

该函数有效，但是有些在第一次迭代后如何停止，我不知道为什么。

Answer 1

您可以将第n个项目存储在单独的变量中，然后将整个列表分配给填充值，然后填充存储的列表（这不会创建原始列表的副本）：

import itertools as it

l = list(range(20))
tmp = l[::5]
l[:] = it.repeat(0, len(l))
l[::5] = tmp

如果可以复印，也可以使用以下方法：

l = list(range(20))
l = [0 if i % 5 else x for i, x in enumerate(l)]

Answer 2

您可以遍历列表，如果索引+ 1不能除以N，则将该项目的值设置为0

data = list(range(1,30))
print(data)
n_param = 5
for i in range(len(data)):
    if (i+1) % n_param:
        data[i] = 0
print(data)

产量

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29]
[0, 0, 0, 0, 5, 0, 0, 0, 0, 10, 0, 0, 0, 0, 15, 0, 0, 0, 0, 20, 0, 0, 0, 0, 25, 0, 0, 0, 0]

Answer 3

您可以使用列表理解来执行此操作，其中x是您的输入列表，n是您要替换的列表的索引

def window(x, n):
   output = [n if (i+1)%n == 0 and i!=0 else 0 for i, item in enumerate(x)]
   return output