Go语言怎么比较两个字符串的相似度-Golang-PHP中文网

Go语言怎么比较两个字符串的相似度

下次还敢

发布： 2025-06-25 16:18:02

原创

469人浏览过

go语言比较字符串相似度的方法包括：1. 编辑距离（levenshtein distance），适用于计算字符差异，使用github.com/agnivade/levenshtein库实现；2. 余弦相似度（cosine similarity），通过词频向量计算相似度，适合长文本；3. jaro-winkler distance，适合短字符串比较，使用github.com/xrash/smetrics库；4. simhash，用于大规模文本快速过滤。选择算法需考虑字符串长度、字符顺序敏感性、计算复杂度和应用场景。优化性能可通过预处理、索引、并行计算和选用高效库实现。对于中文字符串，需进行分词、同义词处理，并选择合适的算法和编码方式。

Go语言怎么比较两个字符串的相似度

Go语言比较两个字符串相似度，核心在于寻找一种量化两者差异的方式。并没有内置的完美函数，但我们可以利用现有的库和算法来实现。

解决方案：

Go语言中比较字符串相似度，可以采用以下几种方法，各有优劣：

立即学习“go语言免费学习笔记（深入）”；

编辑距离（Levenshtein Distance）： 计算将一个字符串转换成另一个字符串所需的最少单字符编辑次数（插入、删除、替换）。距离越小，相似度越高。可以使用github.com/agnivade/levenshtein库。
```
package main

import (
    "fmt"
    "github.com/agnivade/levenshtein"
)

func main() {
    str1 := "kitten"
    str2 := "sitting"
    distance := levenshtein.ComputeDistance(str1, str2)
    fmt.Printf("The Levenshtein distance between '%s' and '%s' is: %d\n", str1, str2, distance)
    // Output: The Levenshtein distance between 'kitten' and 'sitting' is: 3
}
```
登录后复制
编辑距离的优点是简单直观，缺点是计算复杂度较高，且对字符串长度差异敏感。

余弦相似度（Cosine Similarity）： 将字符串视为词频向量，计算两个向量的夹角余弦值。余弦值越接近1，相似度越高。需要先对字符串进行分词和统计词频。可以使用github.com/jbrukh/bayesian库进行简单的分词和分类，然后手动计算余弦相似度。

// 简化示例，需要更完善的分词和向量化处理
package main

import (
    "fmt"
    "math"
    "strings"
)

func cosineSimilarity(str1, str2 string) float64 {
    // 简单的词频统计
    freq1 := make(map[string]int)
    freq2 := make(map[string]int)

    for _, word := range strings.Split(str1, " ") {
        freq1[word]++
    }
    for _, word := range strings.Split(str2, " ") {
        freq2[word]++
    }

    // 计算点积、模长
    dotProduct := 0.0
    magnitude1 := 0.0
    magnitude2 := 0.0

    for word, count := range freq1 {
        dotProduct += float64(count * freq2[word])
        magnitude1 += float64(count * count)
    }

    for _, count := range freq2 {
        magnitude2 += float64(count * count)
    }

    magnitude1 = math.Sqrt(magnitude1)
    magnitude2 = math.Sqrt(magnitude2)

    if magnitude1 == 0 || magnitude2 == 0 {
        return 0.0
    }

    return dotProduct / (magnitude1 * magnitude2)
}

func main() {
    str1 := "this is a foo bar sentence"
    str2 := "this is a foo bar sentence."

    similarity := cosineSimilarity(str1, str2)
    fmt.Printf("Cosine similarity between '%s' and '%s' is: %f\n", str1, str2, similarity)
}

登录后复制

余弦相似度的优点是对字符串长度不敏感，缺点是需要进行分词，且对词序不敏感。

Jaro-Winkler Distance： 专门用于比较短字符串的相似度，考虑了字符匹配和顺序。可以使用github.com/xrash/smetrics库。

package main

import (
    "fmt"
    "github.com/xrash/smetrics"
)

func main() {
    str1 := "MARTHA"
    str2 := "MARHTA"
    distance := smetrics.JaroWinkler(str1, str2, 0.7)
    fmt.Printf("The Jaro-Winkler distance between '%s' and '%s' is: %f\n", str1, str2, distance)
    // Output: The Jaro-Winkler distance between 'MARTHA' and 'MARHTA' is: 0.961111
}

登录后复制

Jaro-Winkler距离的优点是适合短字符串，缺点是对长字符串效果不佳。