【Python】字符串相似性度量，用difflib还是Levenshtein

最新推荐文章于 2025-03-31 17:51:43 发布

兔子爱读书

最新推荐文章于 2025-03-31 17:51:43 发布

阅读量1.2w

点赞数 5

分类专栏： Python编程手册

Python编程手册专栏收录该内容

228 篇文章

订阅专栏

【作者笔记】

difflib（Python自带）：不一定为字符串，数组也可以匹配，但数组匹配时只有单个元素完全匹配才计入相似。

Levenshtein（第三方插件）：需要输入为字符串，匹配时是整体匹配，数组匹配时需要用join把数组元素连接为字符串。

difflib相似度比，和Levenshtein比，基本上接近：

difflib.SequenceMatcher(None, str1,str2).ratio() vs Levenshtein.ratio(str1, str2)

我的感觉是，Levenshtein的原理比较清楚，适合采用。主要是没查到difflib的原理。

我在安装python-Levenshtein时候报错：error: Microsoft Visual C++ 14.0 is required. Get it with "Microsoft Visual C++ Build Tools": http://landinghub.visualstudio.com/visual-cpp-build-tools

成功解决：在这个网址 https://www.lfd.uci.edu/~gohlke/pythonlibs/ 搜索Levenshtein，之后从cmd进入下载目录，用pip install wheel即可安装。

================================================================

字符串的相似性比较应用场合很多，像拼写纠错、文本去重、上下文相似性等。

评价字符串相似度最常见的办法就是：把一个字符串通过插入、删除或替换这样的编辑操作，变成另外一个字符串，所需要的最少编辑次数，这种就是编辑距离（edit distance）度量方法，也称为Levenshtein距离。海明距离是编辑距离的一种特殊情况，只计算等长情况下替换操作的编辑次数，只能应用于两个等长字符串间的距离度量。

其他常用的度量方法还有 Jaccard distance、J-W距离（Jaro–Winkler distance）、余弦相似性（cosine similarity）、欧氏距离（Euclidean distance）等。

方法1：字符串相似度检测的库，difflib

（参考：python字符相似度检验，作者：qcyfred）

query_str = '市公安局'
s1 = '广州市邮政局'
s2 = '广州市公安局'
s3 = '广州市检查院'
print(difflib.SequenceMatcher(None, query_str, s1).quick_ratio())  
print(difflib.SequenceMatcher(None, query_str, s2).quick_ratio())  
print(difflib.SequenceMatcher(None, query_str, s3).quick_ratio())  

# 0.4
# 0.8 --> 某一种相似度评判标准下的最相似的文本……
# 0.08695652173913043

方法2：Levenshtein 使用

（参考：Python 字符串相似性的几种度量方法，作者：-牧野- （CSDN））

莱文斯坦距离（Levenshtein）就用是来解决这个问题的。先看它的公式：

这里写图片描述

其中a，b是两个数组（字符串），i/j是数组下标。莱文斯坦距离的含义，是求将a变成b（或将b变成a），所需要做的最少次数的变换。

举个例子，字符串”kitten”与”sitting”的莱文斯坦距离是3, 应为将一个字符串变为另一个字符串，最小需要做三次变换:

kitten → sitten (字符k变为s)
sitten → sittin (字符e变成i)
sittin → sitting (在末尾插入字符g)、

使用 pip install python-Levenshtein 指令安装 Levenshtein，然后见下面代码示范：

# -*- coding: utf-8 -*-
 
import difflib
# import jieba
import Levenshtein
 
str1 = "我的骨骼雪白 也长不出青稞"
str2 = "雪的日子 我只想到雪中去si"
 
# 1. difflib
seq = difflib.SequenceMatcher(None, str1,str2)
ratio = seq.ratio()
print 'difflib similarity1: ', ratio
 
# difflib 去掉列表中不需要比较的字符
seq = difflib.SequenceMatcher(lambda x: x in ' 我的雪', str1,str2)
ratio = seq.ratio()
print 'difflib similarity2: ', ratio
 
# 2. hamming距离，str1和str2长度必须一致，描述两个等长字串之间对应位置上不同字符的个数
# sim = Levenshtein.hamming(str1, str2)
# print 'hamming similarity: ', sim
 
# 3. 编辑距离，描述由一个字串转化成另一个字串最少的操作次数，在其中的操作包括 插入、删除、替换
sim = Levenshtein.distance(str1, str2)
print 'Levenshtein similarity: ', sim
 
# 4.计算莱文斯坦比
sim = Levenshtein.ratio(str1, str2)
print 'Levenshtein.ratio similarity: ', sim
 
# 5.计算jaro距离
sim = Levenshtein.jaro(str1, str2 )
print 'Levenshtein.jaro similarity: ', sim
 
# 6. Jaro–Winkler距离
sim = Levenshtein.jaro_winkler(str1 , str2 )
print 'Levenshtein.jaro_winkler similarity: ', sim

输出：

difflib similarity1:  0.246575342466
difflib similarity2:  0.0821917808219
Levenshtein similarity:  33
Levenshtein.ratio similarity:  0.27397260274
Levenshtein.jaro similarity:  0.490208958959
Levenshtein.jaro_winkler similarity:  0.490208958959