Use jieba library to identify names of people and places

Design thinking

At first, I directly used Jieba's Feipian mode for tagging, and found that the word segmentation effect was not good, so I first used Jieba to segment the words, and then used Jieba to perform part-of-speech tagging and pass the ranking to Feipian. The effect was quite good.

code

import pandas as pd
import jieba
import jieba.posseg as pseg
jieba.enable_paddle()

def get_per_list(text):
    per_list = []  # 人名列表
    word_list = jieba.lcut(text)
    # print(word_list)
    for word in word_list:
        if len(word)==1:  # 不加判断会爆
            continue
        words = pseg.cut(word, use_paddle=True)  # paddle模式
        # print(list(words))
        word, flag = list(words)[0]
        if flag=='PER':  # 这里写成LOC是地名
            per_list.append(word)
    per_list = list(set(per_list))
    # print(per_list)
    return per_list
text = '张瑞龙毕业于华中农业大学'
per_list = get_per_list(text)
print(per_list)