从列表中取出每个第n个块

Question

给出一个清单:

import string
a = list(string.ascii_lowercase)

什么是返回m个元素的每第n个块的Pythonic方法？请注意,这与仅返回每个第n个元素不同.

获得3个元素的3个块中的每一个的所需结果(取3,跳过6,取3,跳过6 ......):

['a', 'b', 'c', 'j', 'k', 'l', 's', 't', 'u']

我可以这样做:

import itertools
s1 = a[::9]
s2 = a[1::9]
s3 = a[2::9]    
res = list(itertools.chain.from_iterable(zip(s1,s2, s3)))

有更干净的方式吗？

Answer 1

对于选择和跳过的固定顺序，您可以对窗口总长度（此处为 9）取模的索引进行包装，并仅选择低于给定阈值 3 的索引：

lst = [x for i, x in enumerate(a) if i % 9 < 3]
print(lst)
# ['a', 'b', 'c', 'j', 'k', 'l', 's', 't', 'u']

您可以将其变成一个函数，使其使用起来更加直观：

def select_skip(iterable, select, skip):
    return [x for i, x in enumerate(iterable) if i % (select+skip) < select]  

print(select_skip(a, select=3, skip=6))
# ['a', 'b', 'c', 'j', 'k', 'l', 's', 't', 'u']