计算文本相似度并将文件写到文本文件

原创

已于 2022-06-01 20:40:34 修改 · 370 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#文本相似度 #txt文件 #difflib

于 2018-12-30 00:30:28 首次发布

本文介绍了一种使用文本相似度计算来寻找相近颜色的方法，通过difflib库中的SequenceMatcher类计算颜色名称之间的相似度，并将结果保存到txt文件中。

工作中有一个小需求找相近的颜色，就用文本相似度做了，包是直接找的一个包，大概是交集/并集的计算形式，然后将计算结果写到txt文件中

import difflib

A = B = ['纯白',
         '纯黑色',
         '白红色Hi',
         '棕色/白色Hi',
         '白色/蓝色Hi',
         '白色/棕色Hi',
         '白色/蓝色',
         '黑色\白底Hi',
         '米黄色',
         '黑色\白底',
         '白色',
         '深蓝色',
         '大红',
         '白色黑边',
         '布面灰/草绿',
         '深蓝低帮',
         '胭脂红',
         '灰金色',
         '白色',
         '黑武士']

res_list = []

for i, a in enumerate(A, 1):
#第二个参数表示下标开始的位置，取值为1即表示下标从1开始计算，默认从0开始
    for b in B:
        if a != b:
            score = difflib.SequenceMatcher(a=a, b=b).quick_ratio()
            # print(a, b, str(score))
            res_list.append((a, b, str(score)))
    if i % 10 == 0:  # 每10行写入一次
        print(i)
        with open("res.txt", "a", encoding="utf-

最低0.47元/天解锁文章