如何在Python列表中提取指定字符串或索引之间的所有元素？

嘿，我来帮你搞定这个提取列表中两个COMMENT之间元素的需求！针对你给出的示例列表，咱们可以用两种实用思路来实现，下面一步步拆解：

思路1：已知统一分隔符（比如COMMENT） #

如果你的分隔符是固定的同一个字符串（比如这里的COMMENT），咱们可以先找出所有分隔符在列表中的索引位置，再通过相邻索引对来提取中间的元素。

代码实现 #

mylist = ['ABC', 'COMMENT', 'YES', 'YES', 'NO', '123', 'COMMENT','GONOW','MAKE','COMMENT', 'YES','COMMENT']
separator = 'COMMENT'

# 第一步：收集所有分隔符的索引
sep_indices = [i for i, item in enumerate(mylist) if item == separator]

# 第二步：遍历相邻索引对，提取中间元素
results = []
for i in range(len(sep_indices) - 1):
    # 起始位置是分隔符索引+1，结束位置是下一个分隔符的索引
    start_pos = sep_indices[i] + 1
    end_pos = sep_indices[i+1]
    results.append(mylist[start_pos:end_pos])

# 输出结果
for idx, items in enumerate(results, 1):
    print(f"第{idx}个输出：{items}")

运行结果 #

第1个输出：['YES', 'YES', 'NO', '123']
第2个输出：['GONOW', 'MAKE']
第3个输出：['YES']

完全匹配你期望的输出结果～

思路2：通用提取函数（支持指定任意起始/结束标记） #

如果你的需求更灵活——比如要提取不同标记之间的元素，或者指定标记的第N次出现位置，那可以写一个通用函数来处理。

代码实现 #

def get_elements_between(lst, start_str, end_str, start_occurrence=1, end_occurrence=1):
    # 找到起始字符串的指定出现位置的索引
    start_indices = [i for i, item in enumerate(lst) if item == start_str]
    if start_occurrence > len(start_indices):
        return []  # 如果起始标记出现次数不够，返回空列表
    start_idx = start_indices[start_occurrence - 1] + 1

    # 找到结束字符串的指定出现位置（必须在起始标记之后）
    end_indices = [i for i, item in enumerate(lst) if item == end_str and i > start_idx - 1]
    if end_occurrence > len(end_indices):
        return []  # 如果结束标记出现次数不够，返回空列表
    end_idx = end_indices[end_occurrence - 1]

    return lst[start_idx:end_idx]

# 测试示例列表
mylist = ['ABC', 'COMMENT', 'YES', 'YES', 'NO', '123', 'COMMENT','GONOW','MAKE','COMMENT', 'YES','COMMENT']

# 提取第1个COMMENT到第1个后续COMMENT之间的元素
print(get_elements_between(mylist, 'COMMENT', 'COMMENT', 1, 1))
# 提取第2个COMMENT到第1个后续COMMENT之间的元素
print(get_elements_between(mylist, 'COMMENT', 'COMMENT', 2, 1))
# 提取第3个COMMENT到第1个后续COMMENT之间的元素
print(get_elements_between(mylist, 'COMMENT', 'COMMENT', 3, 1))

运行结果 #

['YES', 'YES', 'NO', '123']
['GONOW', 'MAKE']
['YES']

这个函数不仅能处理相同标记的情况，还能应对比如从ABC到第一个COMMENT之间的元素，或者其他任意组合的需求，实用性拉满！

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