数学建模社区-数学中国
标题: 爬虫(六十九)简明 jieba 中文分词教程(六十) [打印本页]
作者: 杨利霞 时间: 2020-5-31 10:40
标题: 爬虫(六十九)简明 jieba 中文分词教程(六十)
爬虫(六十九)简明 jieba 中文分词教程(六十)9 j5 W+ c3 f8 F1 N" L6 l1 _3 G
6 w* O5 ]/ X. c- M
0 引言jieba 是目前最好的 Python 中文分词组件,它主要有以下 3 种特性:
7 C( L5 P6 G: l( M5 S5 @ K1 v
* \$ H' ~* B9 B3 A) \# 导入 jieba
$ R& N8 x& q+ [- Y- Z: g( q9 ?. e. l. K. ?1 o
" b" {4 k& K' p' ]# q9 j1 H; |% a1 p; E
import jieba, R. V& y1 n& P3 p
0 n3 m3 ?/ {9 m5 O" E" L" s+ N6 h
6 ? I' n1 b9 C4 p+ v# G( t% m
: Q5 b, H( a6 y. J- U3 O% }# wimport jieba.posseg as pseg #词性标注0 G0 q( Z' a) l% u% L
. X7 t# E) `* r: u; n# I5 R
- ! x, {* W* T1 {% o H/ }
2 H3 X1 B* K0 z5 G$ {/ C1 U
import jieba.analyse as anls #关键词提取
6 @1 q" T5 y0 H- S* ?! F# o
# V! c7 V. x! a0 _
+ U' r3 L- w; L; y# R, P
1 分词
' X5 [# A* P" L% G$ M: P" H+ p可使用 jieba.cut 和 jieba.cut_for_search 方法进行分词,两者所返回的结构都是一个可迭代的 generator,可使用 for 循环来获得分词后得到的每一个词语(unicode),或者直接使用 jieba.lcut 以及 jieba.lcut_for_search 直接返回 list。其中:* f' O' q3 g; b' X& D0 e1 z
# 尽量不要使用 GBK 字符串,可能无法预料地错误解码成 UTF-8
2 z3 C; q" V( o8 W1.1 全模式和精确模式
& n0 P: S* Q9 J, r5 Z# ^; U- S. r1 R2 t0 W4 H6 ?
$ {- ]3 E' F* }. r) O' O- S4 z1 p9 `$ [
# 全模式
- ^$ O4 ?7 o5 {% g7 q- h( A% t" Z: V% W: S# p R
- & t8 G+ e- z9 }6 i+ F3 ]% e1 b. R
$ p, `7 \. d: A$ A7 vseg_list = jieba.cut("他来到上海交通大学", cut_all=True); C% @' f6 o# b" ^/ [6 q
# V; q3 ]8 q% T( i2 Z. b
- J+ k5 |+ Q% T2 `8 f
' ?* y3 O+ m* Kprint("【全模式】:" + "/ ".join(seg_list))
+ a1 |" J" C3 i2 s5 ~' x
5 M" C+ T9 e7 }. J5 x% ]9 b% _# q6 A$ F: K* q, [
【全模式】:他/ 来到/ 上海/ 上海交通大学/ 交通/ 大学
" `7 B7 f( Q3 R9 y. U% u) C
( o9 m& Z$ |& l( J4 M. P# 精确模式
& M: j+ Z2 A# O0 w+ h3 M; K1 o& x: g$ `; J ?5 o
- 8 V7 f* c9 S* O/ K/ S! K0 f
2 i5 u, R3 v* i7 A8 n E' H
seg_list = jieba.cut("他来到上海交通大学", cut_all=False)
$ Q$ o1 \5 G7 e# t3 {' ~2 _6 N" G3 I4 c* _# r
- & F' k% S3 F- g- _+ j4 e
# D/ p d8 ?# b' |1 r- zprint("【精确模式】:" + "/ ".join(seg_list))8 K1 T1 w- O' B1 Y
7 T, B3 a! L7 y% s+ ^5 S1 i
) Y: T3 ^! W- t# P+ Q& \
【精确模式】:他/ 来到/ 上海交通大学
type(seg_list)generator
- ' T% z& Q6 w F$ I/ r" I6 N+ ^7 ]
8 J) E$ U6 o% }( v: I0 M6 r5 E1 x X
# 返回列表# M+ F+ Q. x) \; c( a N
) u: P8 A' _3 R+ N
- 7 Q w3 a8 [& y4 t5 \
! t7 W3 `7 n) a8 b$ a, _/ kseg_list = jieba.lcut("他来到上海交通大学", cut_all=True)
+ s; j& k+ b, N; U1 w+ j. J1 U4 a1 o& k6 w6 F; w
7 W3 N2 s3 F2 w8 p
* Q4 ^& F- U; e. R3 g/ a9 mprint("【返回列表】:{0}".format(seg_list))1 u+ d6 N: d" h2 f0 t
0 y3 n9 h8 K0 u* Z) |3 O$ F" v4 P! e+ Z! q) B8 i2 t
【返回列表】:['他', '来到', '上海', '上海交通大学', '交通', '大学']
type(seg_list)list
1.2 搜索引擎模式
- " v+ Q/ `4 V6 `! W, @
0 Z% h" Q, D$ W& k
# 搜索引擎模式6 I/ e; G0 n+ C8 T# L
/ X1 p+ o- A, C. o
$ A! V$ o' i3 l3 R; u' Z+ }; C, X- v# F D4 X- o
seg_list = jieba.cut_for_search("他毕业于上海交通大学机电系,后来在一机部上海电器科学研究所工作")
. a+ v, g T- r5 B/ n( C" f- a; r, p+ ^) I7 T; z' X2 D7 |
- 1 G- n' U7 [9 z& [) [) w- G! N: Q
5 ~/ s" Z1 m ^
print("【搜索引擎模式】:" + "/ ".join(seg_list))
, i8 r* G! t; b3 j) z {# Y5 D3 B% |7 O. Q4 {" _3 |; X
7 {- ^. K1 G* |# H2 p! g, l$ U9 J2 U
【搜索引擎模式】:他/ 毕业/ 于/ 上海/ 交通/ 大学/ 上海交通大学/ 机电/ 系/ ,/ 后来/ 在/ 一机部/ 上海/ 电器/ 科学/ 研究/ 研究所/ 工作
4 S" W- c/ N2 ~2 ~% U, z Z: V2 k1 O% @0 R! v, O
# 返回列表
4 {* B2 Z! V! z* _
+ ?4 K8 X) Y) p$ F- i* K- * E# |$ C$ ~/ L
6 k5 E7 Z/ |; O% y6 u6 Useg_list = jieba.lcut_for_search("他毕业于上海交通大学机电系,后来在一机部上海电器科学研究所工作")9 u% V9 E% A" t4 J6 ]
7 ]2 O7 l$ V s. ~3 n
4 O8 K$ b& s9 e7 R& n( g2 t' ~7 q+ n$ M) v; x3 k
print("【返回列表】:{0}".format(seg_list))
( Q4 V P' |5 R" ~$ _) Z( }/ o7 `) \ s$ L* z! o7 j* G% B9 O; v( l
* B# f4 x+ D+ ~) e! V
【返回列表】:['他', '毕业', '于', '上海', '交通', '大学', '上海交通大学', '机电', '系', ',', '后来', '在', '一机部', '上海', '电器', '科学', '研究', '研究所', '工作']
; a# @8 |8 e! _0 _8 n0 h* ~- m- j% F, c+ x C7 ~& c
1.3 HMM 模型
! t0 {: r6 w d7 l# h' @: M3 VHMM 模型,即隐马尔可夫模型(Hidden Markov Model, HMM),是一种基于概率的统计分析模型,用来描述一个系统隐性状态的转移和隐性状态的表现概率。在 jieba 中,对于未登录到词库的词,使用了基于汉字成词能力的 HMM 模型和 Viterbi 算法,其大致原理是:. }5 k' Z' B: G8 D6 V; p, L
采用四个隐含状态,分别表示为单字成词,词组的开头,词组的中间,词组的结尾。通过标注好的分词训练集,可以得到 HMM 的各个参数,然后使用 Viterbi 算法来解释测试集,得到分词结果。( y5 v3 v1 W d# ~7 C$ X
. i* r! G4 g4 ?' |8 I) u
# 代码实现可参考 HmmSeg.py1 \5 K$ E% t" j0 u( Z7 S6 s2 g9 H
5 z8 s" f/ ?) S3 h) O& E7 B* |6 W
* b! n' D9 {3 o S% g" ~
' Z6 n4 y9 P" ?; W( |# r# 未启用 HMM3 p( N: ?7 O) f: ?2 b- i
8 P# g7 [) k6 R0 g9 i& I! Y, R* m- # z2 _: `8 T" P+ v: Y
* F0 L X' V8 r' Z( Jseg_list = jieba.cut("他来到了网易杭研大厦", HMM=False) #默认精确模式和启用 HMM
# X% }0 \& \/ U, N, `+ B8 ?8 s. D$ k( D) D Q/ F8 j z
6 w% J- H8 y$ [/ B; A5 O
3 O# S- @& @. \% Y% @print("【未启用 HMM】:" + "/ ".join(seg_list))
_3 u- A+ ~! K( |) O8 C, k7 \
+ T' x3 d# Y" ?; |
【未启用 HMM】:他/ 来到/ 了/ 网易/ 杭/ 研/ 大厦
- " y& n$ Y4 j8 Q8 j Q6 Q
6 ?& T/ l, @& {7 l5 l3 y# 识别新词! w0 ?" @" B8 Q( b) W8 A6 p
4 Q1 ~: q2 P' V- n; q |
! q9 E$ K& @$ d, ?8 I: q; U c
# f/ ~, e# v7 X. q/ ^seg_list = jieba.cut("他来到了网易杭研大厦") #默认精确模式和启用 HMM
$ X' A8 S( ]$ H6 {8 L4 I3 F; w! g |- R( T7 D( w
- ) O. N8 W7 T* N' X* x" g" {8 B
8 s! I+ ~& m% Y, q0 M7 i% G
print("【识别新词】:" + "/ ".join(seg_list))1 x, c% M6 R& f8 w+ U' r6 g
' q: j# o% y: ]1 b0 [/ @
. H" b2 Y! o" `: s6 q
【识别新词】:他/ 来到/ 了/ 网易/ 杭研/ 大厦
2 繁体字分词jieba 还支持对繁体字进行分词。
/ v4 u5 V. v. R4 [' i b" E) b6 T# a( u8 @5 u) r
# 繁体字文本
: N2 G; u. b/ ]: Y1 k9 z& k( w2 E: N
1 [, l; V% u K; J+ F9 D1 u- ; E' I, g6 y1 @- ^
; e; M; \! v1 b( Y4 z) |2 S
ft_text = """人生易老天難老 歲歲重陽 今又重陽 戰地黃花分外香 壹年壹度秋風勁 不似春光 勝似春光 寥廓江天萬裏霜 """
: y: f& C+ O. R, b2 ^2 K
8 W5 q! n4 ^$ q4 C& W
; ~6 @- p. V. n$ \; q1 l1 R- ]& G8 n: \4 I- e. f7 Y( ^
# 全模式 A0 P# `( E! z8 s4 z
1 {% e9 \, ?- U: W* T& L: N5 P
( _5 h1 F* m6 b+ m3 B0 a
0 k9 j' x4 _% i3 s [; \+ L, S! Y- fprint("【全模式】:" + "/ ".join(jieba.cut(ft_text, cut_all=True))). n2 n4 @ [1 G* P
: A+ Y4 v. Q+ m' ~! ]
0 h1 X& o6 p! G. k0 b0 ^2 V( v
【全模式】:人生/ 易/ 老天/ 難/ 老/ / / 歲/ 歲/ 重/ 陽/ / / 今/ 又/ 重/ 陽/ / / 戰/ 地/ 黃/ 花/ 分外/ 香/ / / 壹年/ 壹/ 度/ 秋/ 風/ 勁/ / / 不似/ 春光/ / / 勝/ 似/ 春光/ / / 寥廓/ 江天/ 萬/ 裏/ 霜/ /6 [2 Y+ w. a4 u7 k7 _. w
' A4 X. N5 E. ?6 N# Z6 K: L
( o/ h) d" i' z6 y# 精确模式, H9 q9 `0 `) Y: H8 j
* w6 l) H; X) N1 d
- " `! }0 _5 G7 ?6 b
$ Q! H! N. @1 V, D
print("【精确模式】:" + "/ ".join(jieba.cut(ft_text, cut_all=False)))4 y8 u5 g; A) a
2 _0 q2 M; _% [: f; ~. C8 M# R
6 d# h+ ~ l. \+ u' T+ i7 ^8 H$ k1 t [
【精确模式】:人生/ 易/ 老天/ 難老/ / 歲/ 歲/ 重陽/ / 今/ 又/ 重陽/ / 戰地/ 黃/ 花/ 分外/ 香/ / 壹年/ 壹度/ 秋風勁/ / 不/ 似/ 春光/ / 勝似/ 春光/ / 寥廓/ 江天/ 萬/ 裏/ 霜/
q [' Q' W1 h9 h
, |1 [+ C8 X$ C- z \% s/ J8 @ f* [/ h
3 f/ y" }2 P) H" q7 N8 y9 S- l% m6 N4 }& y/ Q
# 搜索引擎模式7 x6 {+ Z( _7 E5 b6 i# e" P1 A4 D0 ~
- Q& T& r# }) Z1 g; ` k# r! F8 k
3 b4 Y2 Q! l9 l2 l3 p, e Q5 c
5 I; e2 x {) M+ K( ]; z) V* D; xprint("【搜索引擎模式】:" + "/ ".join(jieba.cut_for_search(ft_text)))0 i, W/ ?$ r7 @( b
C5 |' m1 u2 S& j7 ], G
5 |& t! Z- `0 c3 |
搜索引擎模式】:人生/ 易/ 老天/ 難老/ / 歲/ 歲/ 重陽/ / 今/ 又/ 重陽/ / 戰地/ 黃/ 花/ 分外/ 香/ / 壹年/ 壹度/ 秋風勁/ / 不/ 似/ 春光/ / 勝似/ 春光/ / 寥廓/ 江天/ 萬/ 裏/ 霜/) p- J2 x1 @- V; d# M
1 v5 f: \/ I4 V1 l/ q }# e3 添加自定义词典
- W* B' x" I3 e7 ]2 ^3 y3 N开发者可以指定自定义词典,以便包含 jieba 词库里没有的词,词典格式如下: F, u6 c; y a, a8 s
词语 词频(可省略) 词性(可省略)
6 P6 M% s: F4 f1 s
5 W+ \$ i: m* Q( R例如:/ _! u, C* H/ c7 s4 v9 c
8 y* r! J: E; t8 r6 t! B/ x7 |3 T2 N/ T
- 8 j5 o) n) m( F& X+ X+ ?
N/ p1 E1 t5 V+ ~$ ?9 Z5 ]创新办 3 i: G3 u0 w9 l0 [/ B* j. P, \' K+ t* V
; L7 N# K7 R( s- }5 K8 x# M7 a- v
9 L4 ~9 Y- A& c; h2 \2 T* q9 |6 y/ i& d6 G( x7 m' V
云计算 50 d$ f4 W% O7 G0 d
* Z, B! a/ ^. j4 ?
. a5 Z e' G* g8 z3 Q( N: ^& G6 x$ t w( T
凱特琳 nz
' E6 A, w4 t, k8 u, t
3 |9 o d+ P" n1 j$ n
, R* a; `6 }8 H$ m
# 虽然 jieba 有新词识别能力,但自行添加新词可以保证更高的正确率。
8 z- a0 E) g2 b0 l4 e1 J6 d. n; }3.1 载入词典
3 z# [2 W' C) ^使用 jieba.load_userdict(file_name) 即可载入词典。
; P# X& G, b8 l0 f# file_name 为文件类对象或自定义词典的路径
$ m8 }9 U1 _5 q- N5 a" A/ }
# m" E9 r7 V# E$ {7 p z' r/ c9 Z2 J: a/ B
- # X( F$ D0 W' B
% |$ S' ^1 N- V5 I# 示例文本
, N, R3 u3 G+ y0 k( _: x; D5 n" x" H* h ? U; @1 s% [* D2 x* T
- : X6 p; e) _; W
2 r' N, n5 j9 O$ q$ {7 p1 G' M
sample_text = "周大福是创新办主任也是云计算方面的专家"' H( V" O5 I% a0 R; c
1 \5 d# K$ Y9 S7 j6 z+ Q, k1 O3 r* t; ^) d
- , s) b9 z4 B' u- W& Z1 {
. Y5 {6 R% t. n9 F9 S. S% f+ [ y6 D7 X; n: I7 j0 u
" s7 d6 N" g# j. V0 B7 s6 E
) m" l9 [. _9 w- i% Z( k2 h/ E) h% ^: X" `) Y! S
# 未加载词典& a4 h" T. A' t; r# ?# h- J
( [8 p. N' }: s% g7 U
- ) b2 v _) v. B& i/ H
: K% b5 Y, n0 w% ]print("【未加载词典】:" + '/ '.join(jieba.cut(sample_text)))
a# X/ Z1 U. Z% ]3 R# s/ V) ?; N5 I8 b1 O& F* j ]6 M
3 f( @( O. v& V& |
【未加载词典】:周大福/ 是/ 创新/ 办/ 主任/ 也/ 是/ 云/ 计算/ 方面/ 的/ 专家
【加载词典后】:周大福/ 是/ 创新办/ 主任/ 也/ 是/ 云计算/ 方面/ 的/ 专家/ D' O' \9 }, H3 J. Y# y9 ?* u
. _2 e W( D, M" ?$ _! v3.2 调整词典- o# H6 c3 i7 G" p+ a) e3 I
使用 add_word(word, freq=None, tag=None) 和 del_word(word) 可在程序中动态修改词典。
+ S: }. g# w$ [3 j2 b
( T) k0 B2 j1 M4 v8 u& h/ M4 A& n4 h5 z4 S" x7 a4 }+ C
- % B. [; k/ u' }- e g" ^
2 c' a: i, y4 F2 i- r
jieba.add_word('石墨烯') #增加自定义词语+ }$ I) y' f- f" Y5 [, d) ~
" f M& O/ R6 H* {2 V
6 j M1 L% Z; A- f. m- j) E( W' ?2 E' l. v
jieba.add_word('凱特琳', freq=42, tag='nz') #设置词频和词性: m4 Z) W+ J- k1 B3 `
! I' T8 n" p0 {; O7 X
- 6 _1 L4 ]4 G) m% B i
6 g( S2 B G6 L
jieba.del_word('自定义词') #删除自定义词语# t% j1 s; \1 R+ Q# k$ B8 E
! P$ y' o% s4 t3 P5 x+ r5 K# G1 G
9 k( D$ d) P) w
使用suggest_freq(segment, tune=True)可调节单个词语的词频,使其能(或不能)被分出来
. s0 D8 U7 M. D( V/ X; ?7 g- H' z9 g1 X3 ?- Y/ @: R
# 调节词频前
) R# o& W+ R( Q. ~) e. E- Y
+ f R* d+ ^4 `1 S. N3 {* X; Q
1 t# g& ^$ O& @$ l. Y& k' w0 ?5 w* K" V9 z3 h
print("【调节词频前】:" + '/'.join(jieba.cut('如果放到post中将出错。', HMM=False)))
/ i/ D% N# t g& X- C0 N! G9 |' g3 M- X
' v J% @7 V- F2 k" D$ Z
【调节词频前】:如果/放到/post/中将/出错/。
7 p5 g' r. _4 F$ I" d: P+ i/ d4 } }
# 调节词频
% r+ A& M3 \+ Z/ {# v5 @% d: D& P/ p5 x7 [4 r% p" v
2 {- y+ v. z5 g; ~3 D1 k0 i A' K) G* B5 K h% |0 |- R7 D
jieba.suggest_freq(('中', '将'), True)4 Z3 Z5 I% |7 @( b$ U" ] m
6 K" a7 \/ i: g8 u
, v. J* u F5 }0 a0 `( k
494
- 8 T0 j# T' H8 `% t& n
# M' D/ z+ C& U1 Y0 x# O
# 调节词频后
0 Y5 c9 }! j$ U( Y% L3 @$ y9 X; v: t1 ]; o. A2 S3 B
- , r* i. K1 x$ u2 D
1 c) Q4 s; w3 L7 z1 b6 D& A, T
print("【调节词频后】:" + '/'.join(jieba.cut('如果放到post中将出错。', HMM=False)))9 U+ {! [: ?( D6 p$ S
/ [& Z1 @: K7 h7 ^3 |; j
; ]( t3 e$ i. x& f
【调节词频后】:如果/放到/post/中/将/出错/。
+ t1 o+ A) Z a) a9 |6 H) t
8 j3 _ H5 H3 J: _; A7 M4 关键词提取8 x% d: _* W R
jieba 提供了两种关键词提取方法,分别基于 TF-IDF 算法和 TextRank 算法。1 Y* j5 J3 [7 ]& c7 N
4.1 基于 TF-IDF 算法的关键词提取5 `$ j( B6 ~2 u5 D6 q6 |$ x/ X! V
TF-IDF(Term Frequency-Inverse Document Frequency, 词频-逆文件频率)是一种统计方法,用以评估一个词语对于一个文件集或一个语料库中的一份文件的重要程度,其原理可概括为:
, J/ s& E: c( w! ~3 u一个词语在一篇文章中出现次数越多,同时在所有文档中出现次数越少,越能够代表该文章 ]( r) c( l2 y& k$ z
# k$ O) A! x9 B9 W% J
计算公式:TF-IDF = TF * IDF,其中:% z- j, ]3 O- l. s7 _


通过 jieba.analyse.extract_tags 方法可以基于 TF-IDF 算法进行关键词提取,该方法共有 4 个参数:
0 J3 y% w! l$ G+ ^1 c/ c- G
5 R1 r3 D" e9 i
6 r3 w" A+ o) m4 Xs = "此外,公司拟对全资子公司吉林欧亚置业有限公司增资4.3亿元,增资后,吉林欧亚置业注册资本由7000万元增加到5亿元。吉林欧亚置业主要经营范围为房地产开发及百货零售等业务。目前在建吉林欧亚城市商业综合体项目。2013年,实现营业收入0万元,实现净利润-139.13万元。"5 @2 o$ I! Z+ g4 w1 r5 X$ @ I0 v6 e
! e! k k0 ]3 `8 r* j& j+ g* r' N
" t3 `( a6 U) p. d2 `( D# o! ` X$ v4 E& b: k$ M5 C j
for x, w in anls.extract_tags(s, topK=20, withWeight=True):
4 c4 ]5 Q3 l" x& {( _
; B/ D7 C( ]. I |7 c& F! {7 M* E
9 Y6 a; h* Q1 U7 N
7 \( _; d( w$ O- d y print('%s %s' % (x, w))
4 X1 Q: H. Y) t( h, _. B3 M; ?: j9 p/ j
8 q3 V: j4 Q0 i8 j9 h' P
欧亚 0.7300142700289363
: |. q* Z- V6 S: X9 D! ]吉林 0.6590381843736177 y/ C8 t* e9 f9 Y/ j
置业 0.4887134522112766- Y/ h" d2 g3 B9 p) N: M" q
万元 0.3392722481859574# b' x0 {& v5 W; f3 w4 p
增资 0.335824019852340455 A; p6 `1 Z7 s8 r+ s
4.3 0.25435675538085106" N8 I2 u$ I$ J- {6 |+ L" h
7000 0.254356755380851068 P* w# X8 g( Q& R* q# s/ O( l
2013 0.25435675538085106& j% }3 |0 E5 I. r" {. i
139.13 0.25435675538085106
& e' b5 ~1 i9 v' J I实现 0.199009799003829784 \8 i3 J! n+ [( M2 _
综合体 0.19480309624702127: K8 x* H% V8 [2 f$ ]8 o
经营范围 0.19389757253595744* N( Y# B7 v6 G& ?3 Q
亿元 0.1914421623587234
6 p- w3 ]- Z/ A9 V- E7 F$ H在建 0.17541884768425534: x# |3 V1 ~& \
全资 0.17180164988510638, K5 D9 X( i& p1 i: H7 J7 R' g T9 b
注册资本 0.1712441526
; e$ w% |+ r' Z6 W百货 0.167344600413829793 o# I1 u2 U' v' [" h
零售 0.1475057117057447& P5 G) W+ i7 ^& X; Q& ^4 B* s) W
子公司 0.145960452377872345 N0 D& K, h# T8 D5 U! M: H
营业 0.13920178509021275
* y! d( g2 {0 O$ [( C# M
5 J4 k2 A$ f5 h3 ]2 }使用 jieba.analyse.TFIDF(idf_path=None) 可以新建 TFIDF 实例,其中 idf_path 为 IDF 频率文件。" _4 u/ p( K) Q' q( y
4.2 基于 TextRank 算法的关键词提取
, E: a% w4 [( ^; Z; @9 ITextRank 是另一种关键词提取算法,基于大名鼎鼎的 PageRank,其原理可参见论文—— TextRank: Bringing Order into Texts 。, G2 a1 m' }' ]$ A+ p7 u
通过 jieba.analyse.textrank 方法可以使用基于 TextRank 算法的关键词提取,其与 'jieba.analyse.extract_tags' 有一样的参数,但前者默认过滤词性(allowPOS=('ns', 'n', 'vn', 'v'))。
8 g3 J) T- L2 j) h7 i
8 A6 ]0 d8 @2 ]3 v' T( j, _0 [/ @7 O( i7 x7 S
7 D7 u; L+ \3 O' x8 M- K# }; E
! Y F* i# S; |) p* r5 ffor x, w in anls.textrank(s, withWeight=True):3 i; j8 x5 C& T% D, I0 p& O! e$ b
. c: b2 M5 }: V: S, G Q7 j. n
0 g0 }2 \# g$ E' A+ B @( d5 Z4 U# L( C/ S" {% s' M
print('%s %s' % (x, w))& a0 m4 V1 [+ C, k( D3 i
) {' N: @: Y8 q* n0 z
1 ?' R* f# G% [& R* A! }
吉林 1.0
) I# ~! A& @$ l欧亚 0.99668933541781722 v" N: \8 r# g1 x
置业 0.6434360313092776* ^+ X* [) q; m; x
实现 0.5898606692859626
% _: I& y6 Y9 t! u+ o收入 0.43677859947991454& F4 C2 Y9 Y0 j% `8 f Y/ M9 n1 a
增资 0.40999005312832767 V# {1 A9 {. P
子公司 0.35678295947672795
3 ^' P4 A7 m+ |( D0 j% c城市 0.34971383667403655# @- b$ \1 i2 `6 ^& b
商业 0.34817220716026936
) q, D& X9 H1 {. o0 [业务 0.3092230992619838
) |- `6 Q: s& [# w在建 0.30779291640330888 m) b* t; I4 b @& }5 X8 v
营业 0.3035777049319588
0 b/ y/ ]/ o9 z' Q' u4 n1 T全资 0.303540981053475
m2 ]; X6 w% }: S" i4 R% a综合体 0.29580869172394825- K7 N P$ o7 P- l2 r. H
注册资本 0.29000519464085045! B, X9 D& Y5 ]" h4 ?
有限公司 0.2807830798576574
5 j" t; z6 M/ K: x零售 0.27883620861218145
5 C0 S& A! j) G) n ~5 s百货 0.2781657628445476
7 K3 X/ G6 B/ V5 E' f开发 0.26934887792958515 e" h8 l {! z% @2 w7 G! i' C. [$ x
经营范围 0.2642762173558316+ _8 U, f3 p/ t* M6 y* M3 a2 D; k
( M) i( }. {# T" `! v n; b7 \
使用 jieba.analyse.TextRank() 可以新建自定义 TextRank 实例。
# g" D6 [ F& ~# y3 f* Y4.3 自定义语料库
; p$ ^7 x: S. D关键词提取所使用逆向文件频率(IDF)文本语料库和停止词(Stop Words)文本语料库可以切换成自定义语料库的路径。' |1 ~( q) b) @
, E2 x. V- o$ U9 c* q
2 E! b" A9 ^9 L b
. ~# }' m; o5 \4 t" o0 K' B o
. _, C1 J- }2 ~, F( ~# U0 g$ R% Djieba.analyse.set_stop_words("stop_words.txt") h- i9 d& v8 i$ x5 |! I& ]
8 J- _0 I$ p# u" m( D5 k
- ) E2 W: G3 {: x, \
$ o( I6 R: k l4 {% l% c4 N5 ejieba.analyse.set_idf_path("idf.txt.big");
" e6 B. d5 J( h6 T" `" d: u% l. w" n# \& p U! T& s
- ( O7 ?2 R Q- I* k7 C( {
0 b q, [6 U5 |* n% b! B% a3 v
for x, w in anls.extract_tags(s, topK=20, withWeight=True):: ~; ^$ R- T7 ]8 w% c0 n4 ]8 q
! N6 o( v2 o0 u0 Y: B+ `- w/ Y
* Q. D' @) `1 v# H( M& T5 |+ u9 d% c
print('%s %s' % (x, w))) m/ S& j/ Z: L0 h
) f& f! U" {$ l& n6 V0 s5 E+ a% v
* `5 w9 ?: M+ L5 y& [+ @
吉林 1.0174270215234043/ o0 w! H8 [8 Z" t
欧亚 0.7300142700289363
6 U7 B5 Y+ ]- u/ A$ h& y增资 0.5087135107617021
- E- F: N) B% u2 s实现 0.5087135107617021
& a3 i+ K" T+ @/ Y置业 0.4887134522112766' A, E3 ~; [. }- o: d+ B- \
万元 0.3392722481859574
8 s* D: f3 p1 ^" Q' P# p/ m6 m" W3 w* [此外 0.25435675538085106
, i g" x4 Y# h; t+ L( n全资 0.25435675538085106' _: O0 `8 S! j9 f1 a5 H& o
有限公司 0.25435675538085106+ T$ W/ m' J% m
4.3 0.25435675538085106
' T7 M& L* b- Y2 T B+ T注册资本 0.25435675538085106/ A/ l, \9 {9 x1 N
7000 0.25435675538085106
$ n% X4 `) v `' {& T$ `: L增加 0.25435675538085106/ @/ x0 Z. Y( |# I8 M" R: i
主要 0.25435675538085106
8 t' z9 u6 E _* P房地产 0.25435675538085106
) I7 ]- k. `5 p2 H/ D5 o5 {9 d1 ^业务 0.25435675538085106, p# N" S- w }+ {3 x
目前 0.254356755380851068 m7 ~- \/ M& j4 c) i
城市 0.25435675538085106
$ P* r$ W" W, ^综合体 0.25435675538085106
3 V0 f6 h% i) W; \2013 0.254356755380851063 k. R+ q3 f7 s9 a6 N
) {. c1 A8 s J. l( S6 y% m, L: u
7 @0 L- ?# h/ g# o6 B% R: E6 W5 词性标注
9 l4 P2 C* A8 U7 b' @+ N4 H/ Ajieba.posseg.POSTokenizer(tokenizer=None) 新建自定义分词器,tokenizer 参数可指定内部使用的 jieba.Tokenizer 分词器。jieba.posseg.dt 为默认词性标注分词器。
! N; Z# r( S& v& m/ i% e1 m7 Y# 标注句子分词后每个词的词性,采用和 ictclas 兼容的标记法。
) K9 S/ I* D/ M) F3 `: k) V3 L- B1 e3 b+ A' D
# ]; ^% F" u8 E: j
6 q1 s! r* d7 `2 }9 d
' G3 N3 b0 p! O0 h2 O5 F6 Nwords = pseg.cut("他改变了中国")
0 s- u6 y7 Y1 @8 s6 `! g, ?) Z; |0 T; s6 |2 s
- # P7 w6 A! N7 m* i4 m# n
* x1 w* c3 c: C4 V- p# d3 W: p
for word, flag in words:
8 L7 [: T& P+ s. {5 C0 q# y5 t
' Q) Z4 [& L" `2 q- h
9 O$ E! K, a, M% P print("{0} {1}".format(word, flag))
6 S. z l4 \1 l9 H9 G6 _1 f3 `& f7 ^2 T) e2 B+ w" J
1 @3 d' [. W; K9 o; Q. W, V
他 r
改变 v
了 ul
中国 ns
6 并行分词/ }0 W4 {' |/ J [" m6 ~. u
将目标文本按行分隔后,把各行文本分配到多个 Python 进程并行分词,然后归并结果,从而获得分词速度的可观提升。用法:+ ~# f+ G* @" j$ `. P/ T3 }
# 可参考 test_file.py0 x0 a6 w: x- N5 p' _1 J
注意:基于 python 自带的 multiprocessing 模块,目前暂不支持 Windows- b" T' d2 y" S$ A4 A" R
g; f* z7 R" ~ N' m/ {0 ^
- T; r- o5 g* p$ P u; Q( w$ F5 Q! n7 返回词语在原文的起止位置
使用 jieba.tokenize 方法可以返回词语在原文的起止位置。
注意:输入参数只接受 unicode
- : d" o- G q O, y) x: {
) C# a L2 E+ E! J L0 B9 U
result = jieba.tokenize(u'上海益民食品一厂有限公司')+ }6 h' S f2 r0 G
% Q) M2 b- h9 M9 J. K/ z4 s7 X
9 j! K# Y5 e* h! X9 o5 D0 q- C6 p3 q4 j
print("【普通模式】")" e$ P" j$ R. B. x7 ]8 ~0 E/ x0 _( r/ l
( x( g* l1 b$ l- h
- 7 r4 }' K" `! F( t* m! x
# S. }8 n- Q; `. v
for tk in result:& ?; o2 [ R/ o) W1 Z
0 Z. O' ^! V8 h; g2 d
- 2 B- _/ n7 {/ z! \
u, w! x3 B ~3 N/ S/ B5 [; n: b print("word: {0} \t\t start: {1} \t\t end: {2}".format(tk[0],tk[1],tk[2]))3 c8 T' d6 s1 q* u' R+ }2 X, I$ ]4 Z9 C
" x& B" P% C# p3 ?) D1 A* m2 i& }* k' c
【普通模式】3 u- l7 z6 N6 r5 r
word: 上海 start: 0 end: 2
0 ~2 d1 }: `* ^. i* a, K. U' rword: 益民 start: 2 end: 4
. f+ n1 p" {0 Q# E M- xword: 食品 start: 4 end: 6* A6 i# ]" K& R
word: 一厂 start: 6 end: 8, q) v* d8 F7 e& G& N
word: 有限公司 start: 8 end: 12! C' y( i2 t" y% Y# U( @$ A
& d* I9 }3 n8 ~7 q' Z6 ^/ K
; D4 k' S3 p3 ~# u. ^0 E7 {3 S3 S
( @5 }- G7 N/ T
% H: |' W' x9 W/ ]0 x( a; Kresult = jieba.tokenize(u'上海益民食品一厂有限公司', mode='search')6 q& l( x1 k X7 p1 b" e+ v- P
/ Y! J$ {- A9 y; r" Y, J
- 0 C7 L5 |" R" f3 j& ]! u) e. K
- ~# C* j' u. p$ b7 ^" xprint("【搜索模式】")/ @9 Z, u6 v$ V; a4 B
& y& a3 d, |' j) w7 N2 @9 e
& g0 v- _/ X- d) D- E. {/ ]% d. C
: j& s% [$ T2 ?& |! B' {for tk in result:
2 O8 O: U5 \ Z8 O) u# \$ W1 i7 z2 F$ V- ~8 V, w( x
- % ^( w7 ]) [0 j' v9 a
" g+ Y: p/ H8 O1 D1 [ print("word: {0} \t\t start: {1} \t\t end: {2}".format(tk[0],tk[1],tk[2]))
|" u* c* p+ E) {* U1 v" T0 r7 e/ A
2 X9 C5 j, l; i1 ^4 x" [& Z/ O+ U4 J8 G
【搜索模式】5 R% F( _* c- D) i$ r
word: 上海 start: 0 end: 2
2 e8 S7 ^; \ y1 Z, Kword: 益民 start: 2 end: 4
5 w, w$ r+ V6 K! Z# sword: 食品 start: 4 end: 61 s3 S5 U; `* a
word: 一厂 start: 6 end: 8
# Q/ Q# H. n8 [( s* s2 r" ]: }word: 有限 start: 8 end: 101 |* s" o1 |1 Z
word: 公司 start: 10 end: 12
q9 d3 l. W' E+ G/ L& [word: 有限公司 start: 8 end: 12
3 ~2 M8 g% ?" b+ u% \" _- s7 c* f' R4 r1 e# u1 x
| 欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) |
Powered by Discuz! X2.5 |