如何使用 NLP 和 Python 在字符串中查找城市名称

Question

正如您在标题中看到的，我的问题很简单。我想使用 NLP 查找城市名称。这是一个例子：

问题：

伊斯坦布尔今天天气怎么样？

答案：

["Istanbul"]

我试过这段代码，但对我没有太大帮助。

import nltk   
my_sent = "How is the weather in Izmit"

word = nltk.word_tokenize(my_sent)   
pos_tag = nltk.pos_tag(word)   
chunk = nltk.ne_chunk(pos_tag)   
NE = [ " ".join(w for w, t in ele) for ele in chunk if isinstance(ele, nltk.Tree)]   
print(NE)

我怎样才能得到这个结果？感谢您的回答。

Answer 1

这可以通过spacy 包来完成：

首先你应该安装 spacy 并加载 en_core_web_ln:

python3 -m pip install spacy 
python3 -m spacy download en_core_web_lg

然后使用下面的代码：

import spacy

nlp = spacy.load('en_core_web_lg')

doc = nlp(u'How is the weather in Izmit')

for ent in doc.ents:
    print(ent.text)