有没有能识别一段文字中地名的python代码

识别文本中地名的Python代码通常涉及到自然语言处理（NLP）和地理信息检索技术。一种常见方法是使用预训练的命名实体识别（NER）模型，例如NLTK、spaCy或者Hugging Face的Transformers库中的预训练模型BERT、SpanBERT等。

以下是一个简单的示例，使用spaCy库：

import spacy

# 加载预训练的地名识别模型（这里假设已经安装了`zh_core_web_sm`）
nlp = spacy.load("zh_core_web_sm")

def extract_locations(text):
    doc = nlp(text)
    locations = [ent.text for ent in doc.ents if ent.label_ == "LOC"] # 假设"LOC"是地名标签
    return locations

text = "北京是中国的首都，纽约是美国的城市"
locations = extract_locations(text)
print(locations)  # 输出: ['北京', '纽约']