将SpaCy的EntityRecognizer应用于Pandas数据框中的一列

Question

我有一个基于文本的数据集，我希望将SpaCy的EntityRecognizer应用于特定列的每一行。

我可以通过执行以下操作来应用常规spaCy管道：

df ['new_col'] = df ['col']。apply（lambda x：nlp（x））

我如何仅应用实体确认器并获取其值？

Answer 1

在Spacy中，文档具有属性.ents并在其命名实体上返回生成器。

因此您可以使用：

df['new_col'] = df['col'].apply(lambda x: list(nlp(x).ents))