如何利用集合匹配值获取列表的后续子集

testList 代表从文本文件读取的字符串列表，compareList 代表从字典中获取的字符串列表。需求是找到两者的匹配项后，提取 testList 中该匹配项之后的所有剩余元素到 foundList 中。

现有代码如下：

# Use a set result to print "fox here!"
testList = ['quick', 'brown', 'fox', 'here!']
compareList = ['beige', 'black', 'brown']
foundList = []

result = set(testList) & set(compareList)
# <class 'set'> {'brown'}

解决方法 #

可以通过以下步骤实现需求：

1. 从集合交集结果中取出匹配元素（集合无序，这里用next(iter(result))获取唯一匹配项，若存在多个匹配项可调整逻辑）
2. 找到该元素在testList中的索引位置
3. 对testList进行切片，取索引+1之后的所有元素赋值给foundList

完整代码示例：

testList = ['quick', 'brown', 'fox', 'here!']
compareList = ['beige', 'black', 'brown']
foundList = []

result = set(testList) & set(compareList)
# 取出匹配元素
match = next(iter(result))
# 找到匹配元素在testList中的索引
match_index = testList.index(match)
# 提取匹配项之后的所有元素
foundList = testList[match_index + 1:]

print(foundList)  # 输出: ['fox', 'here!']

扩展说明 #

如果testList中存在多个属于compareList的元素，若需要提取最后一个匹配项之后的元素，可修改索引获取逻辑：

# 获取所有匹配元素在testList中的索引，取最大值（最后一个匹配项的索引）
last_match_index = max(testList.index(item) for item in result)
foundList = testList[last_match_index + 1:]

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