以下是一个示例代码,用于在一个单独的向量中查找并保留出现的词项:
import numpy as np
def keep_matching_words(vector, words):
words_to_keep = []
for word in words:
if word in vector:
words_to_keep.append(word)
return words_to_keep
# 示例输入
vector = np.array(['apple', 'banana', 'cherry', 'apple', 'orange'])
words = ['apple', 'cherry', 'pear']
# 调用函数并打印结果
matching_words = keep_matching_words(vector, words)
print(matching_words)
输出:
['apple', 'cherry']
在这个示例中,我们定义了一个名为keep_matching_words
的函数,它接受一个向量和一个词列表作为输入。它遍历词列表中的每个词,如果该词在向量中出现,则将其添加到一个新的列表words_to_keep
中。最后,函数返回words_to_keep
列表中的词。
在示例输入中,向量包含了一些水果的名称。我们想要保留的词是['apple', 'cherry', 'pear']
。调用keep_matching_words
函数后,返回的匹配词列表为['apple', 'cherry']
,这些词是在向量中出现的。
下一篇:保留xml.etree命名空间