如何在循环中使用数字列表作为索引提取DNA序列指定片段

给定DNA序列列表：

seq_list = ["AATCGATCGG","GACTCGGTTA","AGGGGTACTA"]

需要提取每个序列中索引列表 idx = [7,8,9] 对应的字符，期望输出：

CGG
TTA
CTA

单独指定单个索引（如 for line in seq_list: print(line[7])）可正常运行，但直接使用列表作为索引（for line in seq_list: print(line[idx])）会触发报错：

TypeError: byte indices must be integers or slices, not list
# 或
TypeError: string indices must be integers

解决方案 #

方法1：遍历索引列表拼接字符 #

对每个序列，遍历索引列表中的每个索引，取出对应字符后拼接成字符串输出：

seq_list = ["AATCGATCGG","GACTCGGTTA","AGGGGTACTA"]
idx = [7,8,9]

for seq in seq_list:
    # 列表推导式逐个取字符，再拼接成字符串
    extracted = ''.join([seq[i] for i in idx])
    print(extracted)

方法2：利用切片简化（索引连续时适用） #

由于你的索引[7,8,9]是连续的，可直接用Python的切片语法（切片为左闭右开，因此结束索引设为10）：

seq_list = ["AATCGATCGG","GACTCGGTTA","AGGGGTACTA"]

for seq in seq_list:
    print(seq[7:10])

方法3：使用map函数实现简洁写法 #

通过map将索引取值操作映射到每个索引上，再拼接结果：

seq_list = ["AATCGATCGG","GACTCGGTTA","AGGGGTACTA"]
idx = [7,8,9]

for seq in seq_list:
    print(''.join(map(lambda i: seq[i], idx)))

报错原因说明 #

字符串（或字节串）的索引仅支持单个整数或切片对象，无法直接传入列表作为索引。因此必须逐个取出索引对应的字符，再进行拼接处理。

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