在特定元素出现之前计算列表中的元素数

Question

假设我们有一个存储在这样一个列表中的单词列表：

list_of_words = ["this", "is", "a", "text","for", "demonstration"]

此外，我有一个选定单词的列表：

selected_words = ["hello", "text", "demonstration"]

我想计算在 selected_words 中与 list_of_words 中的单词匹配的单词之前出现在 list_of_words 中的单词（如果您愿意，可以是元素）的数量。

即selected_word 中匹配的第一个单词是“text”，并且在 list_of_words 中该单词之前有 3 个单词。匹配的第二个单词是“demonstration”，前面有 5 个单词。所以输出应该是[3, 5]。

计算这个最有效的方法是什么？我似乎找不到其他人在 python 中问过这个问题。

Answer 1

使用enumerate 和条件list comprehension：

[i for i, word in enumerate(list_of_words) if word in selected_words]
# [3, 5]

如果涉及的列表非常大，您应该考虑事先将selected_words 转换为set 以改进包含检查：

selected_words = set(selected_words)
[i for i, word in enumerate(list_of_words) if word in selected_words]
# [3, 5]