如何在字符串列表中筛选同时包含两个指定单词的元素？

问题场景 #

给定字符串列表，需找出同时包含两个指定单词的元素（可获取对应索引），示例输入：

list_words = ['test cat dog tree',
       'bird time headphone square',
        'cow light gate white', 
        'soccer baseball box volley',
        'clock universe god peace']

word1 = 'gate'
word2 = 'white'

目标：返回索引[2]对应的元素'cow light gate white'

解决方案 #

方法1：基础遍历（兼容索引获取） #

通过enumerate同时获取元素索引和内容，将字符串拆分为单词集合（提升查找效率）后判断匹配：

list_words = ['test cat dog tree',
       'bird time headphone square',
        'cow light gate white', 
        'soccer baseball box volley',
        'clock universe god peace']

word1 = 'gate'
word2 = 'white'

matched_indices = []
matched_elements = []

for idx, string in enumerate(list_words):
    word_set = set(string.split())  # 按空白分割为单词集合
    if word1 in word_set and word2 in word_set:
        matched_indices.append(idx)
        matched_elements.append(string)

print("匹配索引：", matched_indices)  # 输出: [2]
print("匹配元素：", matched_elements)  # 输出: ['cow light gate white']

方法2：列表推导式（简洁实现） #

如果仅需匹配元素，用一行列表推导式即可；若需索引，结合enumerate：

# 仅获取匹配元素
matched_elements = [s for s in list_words if {word1, word2}.issubset(s.split())]

# 同时获取索引和元素
matched_pairs = [(idx, s) for idx, s in enumerate(list_words) if {word1, word2}.issubset(s.split())]

进阶优化 #

• 分隔符适配：若单词用非空白字符分隔（如逗号），将split()改为split(',')
• 大小写忽略：需忽略大小写时，统一转换为小写后判断：

  target_words = {word1.lower(), word2.lower()}
  matched_elements = [s for s in list_words if target_words.issubset(word.lower() for word in s.split())]
  

• 大列表优化：集合的成员查找效率远高于列表，处理大量数据时优先用集合判断

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