Python如何按指定索引提取单层列表中的分组元素？

待处理的原始列表为固定周期排列的单层列表：

li = ['a0','b0','c0','a1','b1','c1',...,'an','bn','cn']

需要拆分得到三个独立分组：['a0',...,'an']、['b0',...,'bn']、['c0',...,'cn']

原代码无法正常运行的核心原因：

• 原列表是单层字符串列表，li[i][k]的写法是对字符串做索引，取到的是单个字符，不是目标元素
• 索引计算逻辑错误，没有匹配列表3个元素为一组的周期排列规律

方案1：匹配原for循环思路的写法 #

列表每3个元素为一个循环单元，同组元素的位置间隔固定为3，只需要设置0、1、2三个起始偏移量，按步长3取值即可：

# 三个分组初始化
group_a, group_b, group_c = [], [], []

for offset in range(3):
    # 从偏移位开始，每次向后跳3个位置取值，刚好覆盖同组所有元素
    current = [li[i] for i in range(offset, len(li), 3)]
    if offset == 0:
        group_a = current
    elif offset == 1:
        group_b = current
    else:
        group_c = current

print(group_a, group_b, group_c, sep='\n')

以n=2的测试用例（即li为['a0','b0','c0','a1','b1','c1','a2','b2','c2']）为例，运行输出：

['a0', 'a1', 'a2']
['b0', 'b1', 'b2']
['c0', 'c1', 'c2']

方案2：更简洁的切片实现 #

Python列表原生支持步长切片，不需要写循环推导式，一行就能拿到对应分组：

group_a = li[::3]   # 从索引0开始，每3个取一个
group_b = li[1::3]  # 从索引1开始，每3个取一个
group_c = li[2::3]  # 从索引2开始，每3个取一个

这个写法和循环逻辑完全一致，执行效率更高，代码更精简。

注意：不要用i*k这类乘法方式计算索引，很容易出现索引越界、取值错位的问题，固定步长遍历是处理周期排列列表最稳妥的方案。

内容的提问来源于stack exchange，提问作者IGY