You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.9 KiB
67 lines
1.9 KiB
from itertools import chain
|
|
from typing import List
|
|
from typing import Text
|
|
|
|
from pypinyin.constants import PHRASES_DICT
|
|
from pypinyin.constants import RE_HANS
|
|
from pypinyin.seg import mmseg
|
|
|
|
|
|
def _seg(chars: Text) -> List[Text]:
|
|
"""按是否是汉字进行分词"""
|
|
s = '' # 保存一个词
|
|
ret = [] # 分词结果
|
|
flag = 0 # 上一个字符是什么? 0: 汉字, 1: 不是汉字
|
|
|
|
for i, c in enumerate(chars):
|
|
if RE_HANS.match(c): # 汉字, 确定 flag 的初始值
|
|
if i == 0: # 第一个字符
|
|
flag = 0
|
|
|
|
if flag == 0:
|
|
s += c
|
|
else: # 上一个字符不是汉字, 分词
|
|
ret.append(s)
|
|
flag = 0
|
|
s = c
|
|
else: # 不是汉字
|
|
if i == 0: # 第一个字符, 确定 flag 的初始值
|
|
flag = 1
|
|
|
|
if flag == 1:
|
|
s += c
|
|
else: # 上一个字符是汉字, 分词
|
|
ret.append(s)
|
|
flag = 1
|
|
s = c
|
|
|
|
ret.append(s) # 最后的词
|
|
return ret
|
|
|
|
|
|
def simple_seg(hans: Text) -> List[Text]:
|
|
"""将传入的字符串按是否是汉字来分割"""
|
|
assert not isinstance(hans, bytes), \
|
|
'must be unicode string or [unicode, ...] list'
|
|
|
|
if isinstance(hans, str):
|
|
return _seg(hans)
|
|
else:
|
|
hans = list(hans)
|
|
if len(hans) == 1:
|
|
return simple_seg(hans[0])
|
|
return list(chain(* [simple_seg(x) for x in hans]))
|
|
|
|
|
|
def seg(hans: Text) -> List[Text]:
|
|
hans = simple_seg(hans)
|
|
ret = []
|
|
for x in hans:
|
|
if not RE_HANS.match(x): # 没有拼音的字符，不再参与二次分词
|
|
ret.append(x)
|
|
elif PHRASES_DICT:
|
|
ret.extend(list(mmseg.seg.cut(x)))
|
|
else: # 禁用了词语库，不分词
|
|
ret.append(x)
|
|
return ret
|