python实体识别代码

以下是一个简单的python实体识别代码：

import nltk
from nltk.tokenize import word_tokenize
from nltk.tag import pos_tag
from nltk.chunk import ne_chunk

# 输入文本
text = "John works at Google in California."

# 分词
tokens = word_tokenize(text)

# 词性标注
pos_tags = pos_tag(tokens)

# 命名实体识别
ne_chunks = ne_chunk(pos_tags)

# 输出结果
for chunk in ne_chunks:
    if hasattr(chunk, 'label'):
        print(chunk.label(), ' '.join(c[0] for c in chunk))

输出结果为：

PERSON John
ORGANIZATION Google
GPE California

这表明在输入的文本中，John被识别为一个人名，Google为一个组织机构名，California为地理位置名。