如何为 python 循环多个追加

How to loop multiple appends for python

num_pixels_per_cell_one_axis = 5
num_cells_per_module_one_axis = 3
inter_cell_sep = 4

max_items_in_list = num_cells_per_module_one_axis * num_pixels_per_cell_one_axis + (num_cells_per_module_one_axis-1) * inter_cell_sep

print(max_items_in_list)

indices_to_retain = list(range(max_items_in_list))
indices_to_remove = indices_to_retain[num_pixels_per_cell_one_axis :: num_pixels_per_cell_one_axis + inter_cell_sep]
if inter_cell_sep == 2:
    for k in range(0,len(indices_to_remove)):
        indices_to_remove.append(indices_to_remove[k]+1)
        
if inter_cell_sep == 3:
    for k in range(0,len(indices_to_remove)):
        indices_to_remove.append(indices_to_remove[k]+1)
        indices_to_remove.append(indices_to_remove[k]+2)
        
    
for k in indices_to_remove:
    indices_to_retain.remove(k)
    

print(indices_to_remove)
print(indices_to_retain)

我想找到一种方法来循环 inter_cell_sep 任何正数,并且随着它的增加,用于附加列表的行也会增加。预期的答案应该是 [0,1,2,3,4,9,10,11,12,17,18,19,20]

我认为您可以遍历 inter_cell_sep 的范围,而不是对 inter_cell_sep 的每个值使用 if 语句。这是我想出的。

num_pixels_per_cell_one_axis = 5
num_cells_per_module_one_axis = 3
inter_cell_sep = 3

max_items_in_list = num_cells_per_module_one_axis * num_pixels_per_cell_one_axis + (num_cells_per_module_one_axis - 1) * inter_cell_sep

print(max_items_in_list)

indices_to_retain = list(range(max_items_in_list))
indices_to_remove = indices_to_retain[num_pixels_per_cell_one_axis:: num_pixels_per_cell_one_axis + inter_cell_sep]

for k in range(0, len(indices_to_remove)):
    indices_to_remove.extend([indices_to_remove[k] + x for x in range(1, inter_cell_sep)])

for k in indices_to_remove:
    indices_to_retain.remove(k)

print(indices_to_remove)
print(indices_to_retain)