安装

  • 全自动:easy_install jieba 或者 pip install jieba 或者pip3 install jieba
  • 手 动:jieba 目录放置于当前目录或者 site-packages 目录
  • 半自动:下载http://pypi.python.org/pypi/jieba/、解压、运行python setup.py install

算法

  • 基于统计词典,构造前缀词典;基于前缀词典对句子进行切分,得到所有切分可能;根据切分位置,构造一个有向无环图(DAG)
  • 基于DAG图,采用动态规划计算最大概率路径(最有可能的分词结果),根据最大概率路径分词
  • 对于未登录词(词库中没有的词),采用有汉字成词能力的 HMM 模型进行切分
  • 注: jieba中没有使用trie树作为前缀词典存储的数据结构,用的是前缀集合,如set([‘数’, ‘数据’, ‘数据结’, ‘数据结构’])

功能

  1、分词(三种分词模式 )

  A、精确模式:将句子精确切开,适合文本分析。默认是精确模式。 
  B、全模式:将句中所有可以成词的词语都扫描出来,适合用于搜索引擎构建倒排索引的分词,粒度比较细 
  C、搜索引擎模式:在精确模式基础上,对长词再次切分,适合搜索引擎 
  注:当指定jieba.cut_for_search或jieba.cut的参数HMM=True时,就有了新词发现的能力

# encoding=utf-8

import jieba

# 精确模式

seg_list = jieba.cut("我去过清华大学和北京大学。")
print(unicode("精确模式: ","utf-8")+ "/ ".join(seg_list))

# 全模式

seg_list = jieba.cut("我去过清华大学和北京大学。", cut_all=True)
print(unicode("全模式: ","utf-8") + "/ ".join(seg_list))

# 搜索引擎模式

seg_list = jieba.cut_for_search("我去过清华大学和北京大学。")
print(unicode("搜索引擎模式: ","utf-8") + "/ ".join(seg_list))

# 精确模式/全模式下-新词发现 “杭研”没有在词典中,也被HMM模型 Viterbi算法识别出来

seg_list = jieba.cut("他来到了网易杭研大厦",HMM=True)
print(unicode("精确模式/全模式-新词发现: ","utf-8") + "/ ".join(seg_list))

# 搜索引擎模式下-新词发现 “杭研”没有在词典中,也被HMM模型 Viterbi算法识别出来

seg_list = jieba.cut_for_search("他来到了网易杭研大厦",HMM=True)
print(unicode("搜索引擎模式-新词发现: ","utf-8") + "/ ".join(seg_list))

#print result:

#精确模式: 我/ 去过/ 清华大学/ 和/ 北京大学/ 。

#全模式: 我/ 去过/ 清华/ 清华大学/ 华大/ 大学/ 和/ 北京/ 北京大学/ 大学/ /

#搜索引擎模式: 我/ 去过/ 清华/ 华大/ 大学/ 清华大学/ 和/ 北京/ 大学/ 北京大学/ 。

#精确模式/全模式-新词发现: 他/ 来到/ 了/ 网易/ 杭研/ 大厦

#搜索引擎模式-新词发现: 他/ 来到/ 了/ 网易/ 杭研/ 大厦

2、添加自定义用户词典

      Jieba 有新词发现能力,但添加用户词典可以保证更高的正确率。 

      无用户词典下中英文分词示例

# encoding=utf-8

import jieba

data=[
   "世界经济论坛也叫达沃斯论坛。",
   "The World Economic Forum is also called the Davos Forum."
    ]

for d in data:
   seg_list = jieba.cut(d)
   #词与词之间用","连接
   print(",".join(seg_list))

#无用户词典下中英文分词结果

   # 世界, 经济, 论坛, 也, 叫, 达沃斯, 论坛,。
   # The,, World,, Economic,, Forum,, is, , also,, called,, the,, >Davos,, Forum,.

    有用户词典下的中英文分词

# encoding=utf-8
import jieba

"""
默认格式下的用户词典"dict/userdict:
世界经济论坛 n
达沃斯论坛 n
World Economic Forum n
Davos Forum n

"""

# 加载用户词典

jieba.load_userdict("dict/userdict")

# 测试用户词典

data=[
   "世界经济论坛也叫达沃斯论坛。",
   "The World Economic Forum is also called the Davos Forum."
    ]

for d in data:
   seg_list = jieba.cut(d)
   #词与词之间用","连接
   print(",".join(seg_list))

#默认格式用户词典下中英文分词结果

   # 世界经济论坛,也,叫,达沃斯论坛,。
   # The, ,World, ,Economic, ,Forum, ,is, ,also, ,called, ,the, ,Davos, >,Forum,.

可以看到jieba分词在默认格式下,中文没问题,英文专有名词不太理想。不过,可以通过修改jieba包中的init.py文件几个正则来解决。
  修改格式后用户词典下中英文分词示例 
 

# encoding=utf-8
import jieba

"""
修改格式后的用户词典"dict/userdict:用@@来间隔词、词频、词性
世界经济论坛@@n
达沃斯论坛@@n
World Economic Forum@@n
Davos Forum@@n

"""

# 加载用户词典

jieba.load_userdict("dict/userdict")

# 测试用户词典

data=[
   "世界经济论坛也叫达沃斯论坛。",
   "The World Economic Forum is also called the Davos Forum."
    ]

for d in data:
   seg_list = jieba.cut(d)
   #词与词之间用","连接
   print(",".join(seg_list))

#修改格式后中英文分词结果

   # 世界经济论坛,也,叫,达沃斯论坛,。
   # The, ,World Economic Forum, ,is, ,also, ,called, ,the, ,Davos >Forum,.

3、关键词提取

   基于 TF-IDF 算法的关键词抽取 
    jieba.analyse.extract_tags(sentence, topK=20, withWeight=False, allowPOS=(‘ns’, ‘n’, ‘vn’, ‘v’)) 
    参数解释: 
      (1)sentence :待提取的文本; 
      (2)topK :返回K个权重最大的关键词 
      (3)withWeight :是否返回权重 
      (4)allowPOS :是否仅包括指定词性的词
 

# encoding=utf-8

#关键词提取

import jieba.analyse

kWords = jieba.analyse.extract_tags("此外,公司拟对全资子公司吉林欧亚置业>有限公司增资4.3亿元,增资后,吉林欧亚置业注册资本由7000万元增加到5亿元。>吉林欧亚置业主要经营范围为房地产开发及百货零售等业务。目前在建吉林欧亚城>市商业综合体项目。2013年,实现营业收入0万元,实现净利润-139.13万元。", >topK=5,withWeight=True)
for word,weight in kWords:
   # print(word+":"+weight)
   print word,":",weight

#结果

# 欧亚 : 0.730014270029
# 吉林 : 0.659038184374
# 置业 : 0.488713452211
# 万元 : 0.339272248186
# 增资 : 0.335824019852


"""
注: 关键词提取所使用逆文频(IDF)库、停用词库可切换成自定路径 
基于 TextRank 算法的关键词抽取 
   jieba.analyse.textrank(sentence, topK=20, withWeight=False, allowPOS=(‘ns’, ‘n’, ‘vn’, ‘v’)) 
     参数解释: 
       (1)sentence :待提取的文本; 
       (2)topK :返回K个权重最大的关键词 
       (3)withWeight :是否返回权重 
       (4)allowPOS :是否仅包括指定词性的词 
     TextRank基本思想: 
       (1)分词 
       (2)找词之间共现关系,构建图 
       (3)计算图中节点的PageRank
"""

4、词性标注

import jieba.posseg as pseg
words = pseg.cut("我爱北京天安门")
for word, flag in words:
print('%s %s' % (word, flag))

#词性
#我 r
#爱 v
#北京 ns
#天安门 ns

5、并行分词

      原理: 
    文本按行分隔后,分配到多个 Python 进程并行分词,最后归并结果。 
   用法: 
    jieba.enable_parallel(4) # 开启并行分词模式,参数为并行进程数 
    jieba.disable_parallel() # 关闭并行分词模式 
   效果: 
    在 4 核 3.4GHz Linux 机器上,对金庸全集进行精确分词,获得了 1MB/s 的速度,是单进程版的 3.3 倍。 
    注意: 
    并行分词仅支持默认分词器 jieba.dt 和 jieba.posseg.dt。目前暂不支持 Windows
6、返回词语在原文的起止位置

import jieba
result = jieba.tokenize(u'永和服装饰品有限公司')
for tk in result:
   print("word %s\t\t start: %d \t\t end:%d" % (tk[0],tk[1],tk[2]))

#结果

#word 永和       start: 0        end:2
#word 服装       start: 2        end:4
#word 饰品       start: 4        end:6
#word 有限公司     start: 6        end:10
#word 不是       start: 10       end:12