使用短语中的信息查找句子中单词的索引

Question

我需要sentence 中的word 的索引。但有时会出现重复的单词。 phrase 信息会很有帮助。或word 列中的上一行或下一行。

基本上，我只需要识别话语中的单词，例如如果word 是“海边”，我想知道它在句子中是哪个“海边”。我有来自phrase 的额外信息可以帮助识别。它们在数据框中的出现顺序也有帮助。

我现在拥有的是这样的：

file_id	phrase	word	sentence	word_indices
A	I am	I	I am a happy bird. I sing every day. I eat worms.	[0, 5, 9]
B	the seaside is	the	she is by the seaside. The seaside is packed.	[3, 5]
B	the seaside is	seaside	she is by the seaside. The seaside is packed.	[4, 6]
B	the seaside is	is	she is by the seaside. The seaside is packed.	[1, 7]
C	nobody knows	nobody	nobody knows what is going on. She can find nobody	[0, 9]
C	find nobody	nobody	nobody knows what is going on. She can find nobody	[0, 9]
D	it is such a sunny day	sunny	it is such a sunny day ah I am so happy when it's sunny such a sunny day is the best	[4, 13, 16]

但我想得到的是target 列中的内容。

file_id	phrase	word	sentence	word_indices	target
A	I am	I	I am a happy bird. I sing every day. I eat worms.	[0, 5, 9]	[0]
B	the seaside is	the	she is by the seaside. The seaside is packed.	[3, 5]	[5]
B	the seaside is	seaside	she is by the seaside. The seaside is packed.	[4, 6]	[6]
B	the seaside is	is	she is by the seaside. The seaside is packed.	[1, 7]	[7]
C	nobody knows	nobody	nobody knows what is going on. She can find nobody	[0, 9]	[0]
C	find nobody	nobody	nobody knows what is going on. She can find nobody	[0, 9]	[9]
D	it is such a sunny day	sunny	it is such a sunny day ah I am so happy when it's sunny such a sunny day is the best	[4, 13, 16]	[4]

我在这里发现了一个类似的问题：Find index of words in matched text 但不幸的是，这是在 java 中，我需要使用 python 来回答。

非常感谢！

Answer 1

我会将其分为两个步骤。找出导致句子中短语的单词数，然后找到短语中单词的单词索引号：如下所示：

def get_index_of_word_in_sentence(word, phrase, sentence):
    index1 = sentence.index(phrase)
    word_num1 = len(sentence[:index1].split())
    word_num2 = phrase.split().index(word)
    return word_num1 + word_num2

df["target"] = df.apply(lambda x: get_index_of_word_in_sentence(x["word"], x["phrase"], x["sentence"]), axis=1)