Python 實現語句中提取人名（附代碼） | Python工具

前言

本文提供將語句中的人名提取出來的工具方法，可以拿去直接使用。

環境依賴

需要安裝兩個庫，其實一個也可以，但是我這邊准備了兩個庫做個比較。

安裝命令如下：

pip install LAC -i https://pypi.douban.com/simple
pip install ltp -i https://pypi.douban.com/simple

代碼

不廢話，上代碼。

#!/user/bin/env python
# coding=utf-8
"""
@project : csdn
@author  : 劍客阿良_ALiang
@file   : extract_sentence_name_tool.py
@ide    : PyCharm
@time   : 2022-01-25 11:11:43
"""
from LAC import LAC
from ltp import LTP
import time

lac = LAC(mode="lac")
ltp = LTP()


# 句子提取名字
def extract_name(sentence: str, type='lac'):
    user_name_lis = []
    if type == 'lac':
        _result = lac.run(sentence)
        for _index, _label in enumerate(_result[1]):
            if _label == "PER":
                user_name_lis.append(_result[0][_index])
    elif type == 'ltp':
        _seg, _hidden = ltp.seg([sentence])
        _pos_hidden = ltp.pos(_hidden)
        for _seg_i, _seg_v in enumerate(_seg):
            _hidden_v = _pos_hidden[_seg_i]
            for _h_i, _h_v in enumerate(_hidden_v):
                if _h_v == "nh":
                    user_name_lis.append(_seg_v[_h_i])
    else:
        raise Exception('type not suppose')
    return user_name_lis


if __name__ == '__main__':
    _start_lac = time.time()
    lis1 = extract_name("就因為看了沈騰和賈玲的王牌對王牌節目，所以楊迪肯定偷題了。", 'lac')
    _end_lac = time.time()
    print("LAC: {} <- 耗時[{}]秒".format(lis1, (_end_lac - _start_lac)))

    _start_ltp = time.time()
    lis2 = extract_name("就因為看了沈騰和賈玲的王牌對王牌節目，所以楊迪肯定偷題了。", 'ltp')
    _end_ltp = time.time()
    print("LTP: {} <- 耗時[{}]秒".format(lis2, (_end_ltp - _start_ltp)))

代碼說明：

1、extract_name方法入參分別為：語句參數、類型參數。其中默認為lac模式，可以選擇ltp模式。

其中lac模型提取人名的速率較快，但是ltp的提取人名准確率更高。

驗證一下，執行看看效率。

總結

使用的時候可以多試試兩個庫的區別，ltp的准確率稍微高一點。

分享：每個人都睜着眼睛，但不等於每個人都在看世界，許多人幾乎不用自己的眼睛看，他們只聽別人說，他們看到的世界永遠是別人說的樣子。——猜猜看

如果本文對你有幫助的話，點個贊吧，謝謝！

本人CSDN主頁地址:劍客阿良_ALiang的主頁

一起學習，一起進步。