Як порівняти рядки в GoLang?


89

Я не можу отримати "справжній" результат, коли йдеться про порівняння рядків Go. Я написав наступне, щоб пояснити проблему, і додав знімок екрана

// string comparison in Go
package main
import "fmt"
import "bufio"
import "os"

func main() {
    var isLetterA bool 

    fmt.Println("Enter the letter a")
    reader := bufio.NewReader(os.Stdin)
    input, _ := reader.ReadString('\n')

    if(input == "a") {
        isLetterA = true
    } else {
        isLetterA = false 
    }

    fmt.Println("You entered",input)
    fmt.Println("Is it the letter a?",isLetterA)

}

приклад


користувачі Windows перевіряють мою відповідь :)
Дакш Міглані

Відповіді:


128

==є правильним оператором для порівняння рядків у Go. Однак рядки, з якими ви читаєте STDIN reader.ReadString, не містять "a", але "a\n"(якщо ви уважно подивитесь, ви побачите додатковий розрив рядка у своєму прикладі виводу).

Ви можете використовувати strings.TrimRightфункцію, щоб видалити кінцеві пробіли з вашого вводу:

if strings.TrimRight(input, "\n") == "a" {
    // ...
}

9

Для незалежних користувачів від платформи або користувачів Windows ви можете:

час виконання імпорту:

import (
    "runtime"
    "strings"
)

а потім обріжте рядок так:

if runtime.GOOS == "windows" {
  input = strings.TrimRight(input, "\r\n")
} else {
  input = strings.TrimRight(input, "\n")
}

тепер ви можете порівняти це так:

if strings.Compare(input, "a") == 0 {
  //....yourCode
}

Це кращий підхід, коли ви використовуєте STDIN на кількох платформах.

Пояснення

Це трапляється тому, що на вікнах рядки, кінці "\r\n"яких відомі як CRLF, а на рядках UNIX, які закінчуються на те, "\n"що називається LF, і тому ми "\n"обробляємо операційні системи на основі Unix, тоді як обрізаємо "\r\n"на вікнах.


4
Не потрібно диференціювати. Другим аргументом є набір, а не суфікс, і будь-які символи в наборі будуть обрізані у будь-якому порядку / комбінації. Обрізання "\ r \ n" буде достатньо для обох.
Джейсон Карлсон,

1

Якщо припустити, що немає попередніх / наступних пробілів, все ще існує кілька способів ствердження рівності рядків. Деякі з них:

Ось декілька основних результатів тесту (у цих тестах, strings.EqualFold(.., ..)здається, найбільш ефективний вибір):

goos: darwin
goarch: amd64
BenchmarkStringOps/both_strings_equal::equality_op-4               10000        182944 ns/op
BenchmarkStringOps/both_strings_equal::strings_equal_fold-4        10000        114371 ns/op
BenchmarkStringOps/both_strings_equal::fold_caser-4                10000       2599013 ns/op
BenchmarkStringOps/both_strings_equal::lower_caser-4               10000       3592486 ns/op

BenchmarkStringOps/one_string_in_caps::equality_op-4               10000        417780 ns/op
BenchmarkStringOps/one_string_in_caps::strings_equal_fold-4        10000        153509 ns/op
BenchmarkStringOps/one_string_in_caps::fold_caser-4                10000       3039782 ns/op
BenchmarkStringOps/one_string_in_caps::lower_caser-4               10000       3861189 ns/op

BenchmarkStringOps/weird_casing_situation::equality_op-4           10000        619104 ns/op
BenchmarkStringOps/weird_casing_situation::strings_equal_fold-4    10000        148489 ns/op
BenchmarkStringOps/weird_casing_situation::fold_caser-4            10000       3603943 ns/op
BenchmarkStringOps/weird_casing_situation::lower_caser-4           10000       3637832 ns/op

Оскільки варіантів досить багато, тож ось код для створення тестів.

package main

import (
    "fmt"
    "strings"
    "testing"

    "golang.org/x/text/cases"
    "golang.org/x/text/language"
)

func BenchmarkStringOps(b *testing.B) {
    foldCaser := cases.Fold()
    lowerCaser := cases.Lower(language.English)

    tests := []struct{
        description string
        first, second string
    }{
        {
            description: "both strings equal",
            first: "aaaa",
            second: "aaaa",
        },
        {
            description: "one string in caps",
            first: "aaaa",
            second: "AAAA",
        },
        {
            description: "weird casing situation",
            first: "aAaA",
            second: "AaAa",
        },
    }

    for _, tt := range tests {
        b.Run(fmt.Sprintf("%s::equality op", tt.description), func(b *testing.B) {
            for i := 0; i < b.N; i++ {
                benchmarkStringEqualsOperation(tt.first, tt.second, b)
            }
        })

        b.Run(fmt.Sprintf("%s::strings equal fold", tt.description), func(b *testing.B) {
            for i := 0; i < b.N; i++ {
                benchmarkStringsEqualFold(tt.first, tt.second, b)
            }
        })

        b.Run(fmt.Sprintf("%s::fold caser", tt.description), func(b *testing.B) {
            for i := 0; i < b.N; i++ {
                benchmarkStringsFoldCaser(tt.first, tt.second, foldCaser, b)
            }
        })

        b.Run(fmt.Sprintf("%s::lower caser", tt.description), func(b *testing.B) {
            for i := 0; i < b.N; i++ {
                benchmarkStringsLowerCaser(tt.first, tt.second, lowerCaser, b)
            }
        })
    }
}

func benchmarkStringEqualsOperation(first, second string, b *testing.B) {
    for n := 0; n < b.N; n++ {
        _ = strings.ToLower(first) == strings.ToLower(second)
    }
}

func benchmarkStringsEqualFold(first, second string, b *testing.B) {
    for n := 0; n < b.N; n++ {
        _ = strings.EqualFold(first, second)
    }
}

func benchmarkStringsFoldCaser(first, second string, caser cases.Caser, b *testing.B) {
    for n := 0; n < b.N; n++ {
        _ = caser.String(first) == caser.String(second)
    }
}

func benchmarkStringsLowerCaser(first, second string, caser cases.Caser, b *testing.B) {
    for n := 0; n < b.N; n++ {
        _ = caser.String(first) == caser.String(second)
    }
}

0

Вміст усередині рядків у Golang можна порівняти за допомогою ==оператора. Якщо результати не як очікувалося можуть бути деякі приховані символи , такі як \n, \rпрогалини і т.д. Так як загальне правило, спробуйте видалити ті , які використовують функції , що надаються stringsпакетом в golang.

Наприклад, пробіли можна видалити за допомогою strings.TrimSpaceфункції. Ви також можете визначити власну функцію, щоб видалити будь-який потрібний вам символ. strings.TrimFuncфункція може дати вам більше енергії.

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.