根据列表项的索引模数执行不同的操作

Question

对于给定的列表，我希望：

1. 仅对第一个元素，然后是每第三个后续元素（索引 3、6 等），什么都不做
2. 从第二个元素（索引 1）开始，然后对于每 3 个元素（索引 4、7 等），执行特定操作
3. 从第三个元素（索引 2）开始，然后每 3 个元素（索引 5、8 等）执行不同的操作

我可以使用组合 range(len(mylist)) 和 :: 表示法来逐步分割列表：

1::3 will give me the elements at indices 1, 4, 7, 10 etc.

::2 will give me the elements at indices 0, 2, 4, 6 etc.

但是这里有几个问题：

• 我不需要访问元素 6，例如
• 有一些重叠 两片之间
• 缺少 5 等索引

Python 是否为自定义逐步切片提供内置机制？ FWIW，我实际上是在处理一个pandas（0.18.0版）系列，为了便于解释，我只是使用了一个常规列表的例子。

Answer 1

您不需要对数据进行多次传递，或进行任何切片：

def nop(x): return x

def op1(x): return x.upper()

def op2(x): return x[::-1]

data = 'One Two Three Four Five Six Seven Eight Nine'.split()

result = [(nop, op1, op2)[i % 3](x) for i, x in enumerate(data)]

这会产生以下result：

['One', 'TWO', 'eerhT', 'Four', 'FIVE', 'xiS', 'Seven', 'EIGHT', 'eniN']

如果不清楚，这里发生的是 (nop, op1, op2) 的第 (i%3) 个元素以 x 作为参数调用，对于每个索引 i 和元素 x data的枚举。

显然，如果您需要对data 执行操作而不是返回新列表，您可以在常规的for 循环中执行此操作：

for i, x in enumerate(data):
    data[i] = (nop, op1, op2)[i % 3](x)