Converter string para o tipo inteiro no Go?


236

Estou tentando converter uma string retornada de flag.Arg(n)para uma int. Qual é a maneira idiomática de fazer isso no Go?

Respostas:


298

Por exemplo,

package main

import (
    "flag"
    "fmt"
    "os"
    "strconv"
)

func main() {
    flag.Parse()
    s := flag.Arg(0)
    // string to int
    i, err := strconv.Atoi(s)
    if err != nil {
        // handle error
        fmt.Println(err)
        os.Exit(2)
    }
    fmt.Println(s, i)
}

14
func main() { ... }não aceita argumentos e não retorna valor. Use a função ospackage Exit, por exemplo:os.Exit(2).
peterSO 26/11

2
Como alternativa, basta fazer um fatal Egpanic(err)
Peter Bengtsson

70

Convertendo cadeias simples

A maneira mais fácil é usar a strconv.Atoi()função.

Observe que existem muitas outras maneiras. Por exemplo, fmt.Sscan()e strconv.ParseInt()que oferecem maior flexibilidade, você pode especificar a base e o tamanho de bits, por exemplo. Também conforme observado na documentação de strconv.Atoi():

Atoi é equivalente a ParseInt (s, 10, 0), convertido para o tipo int.

Aqui está um exemplo usando as funções mencionadas (experimente no Go Playground ):

flag.Parse()
s := flag.Arg(0)

if i, err := strconv.Atoi(s); err == nil {
    fmt.Printf("i=%d, type: %T\n", i, i)
}

if i, err := strconv.ParseInt(s, 10, 64); err == nil {
    fmt.Printf("i=%d, type: %T\n", i, i)
}

var i int
if _, err := fmt.Sscan(s, &i); err == nil {
    fmt.Printf("i=%d, type: %T\n", i, i)
}

Saída (se chamada com argumento "123"):

i=123, type: int
i=123, type: int64
i=123, type: int

Analisando strings personalizadas

Há também um recurso útil fmt.Sscanf()que oferece flexibilidade ainda maior, pois, com a sequência de formatação, você pode especificar o formato numérico (como largura, base etc.) junto com caracteres extras adicionais na entrada string.

Isso é ótimo para analisar seqüências personalizadas mantendo um número. Por exemplo, se sua entrada é fornecida em uma forma em "id:00123"que você tem um prefixo "id:"e o número é fixado em 5 dígitos, preenchido com zeros se for menor, isso é muito facilmente analisável assim:

s := "id:00123"

var i int
if _, err := fmt.Sscanf(s, "id:%5d", &i); err == nil {
    fmt.Println(i) // Outputs 123
}

O que o segundo argumento ParseIntespecifica?
kaushik94

1
@ kaushik94 Clique no strconv.ParseInt()link e você verá imediatamente: ParseInt(s string, base int, bitSize int). Portanto, é a base: "ParseInt interpreta uma string s na base fornecida (2 a 36)"
icza

Observe que o argumento bitSize em strconv.ParseInt () não converterá a string na sua escolha de tipo, mas apenas existe para limitar o resultado a uma 'testemunha' específica. Veja também: stackoverflow.com/questions/55925894/…
viv

@viv Sim, está correto. Se um valor do tipo intfor necessário e strconv.ParseInt()for usado, será necessária a conversão manual do tipo (de int64para int).
icza 24/06/19

16

Aqui estão três maneiras de analisar seqüências de caracteres em números inteiros, do tempo de execução mais rápido ao mais lento:

  1. strconv.ParseInt(...) o mais rápido
  2. strconv.Atoi(...) ainda muito rápido
  3. fmt.Sscanf(...) não terrivelmente rápido, mas mais flexível

Aqui está uma referência que mostra o tempo de uso e exemplo para cada função:

package main

import "fmt"
import "strconv"
import "testing"

var num = 123456
var numstr = "123456"

func BenchmarkStrconvParseInt(b *testing.B) {
  num64 := int64(num)
  for i := 0; i < b.N; i++ {
    x, err := strconv.ParseInt(numstr, 10, 64)
    if x != num64 || err != nil {
      b.Error(err)
    }
  }
}

func BenchmarkAtoi(b *testing.B) {
  for i := 0; i < b.N; i++ {
    x, err := strconv.Atoi(numstr)
    if x != num || err != nil {
      b.Error(err)
    }
  }
}

func BenchmarkFmtSscan(b *testing.B) {
  for i := 0; i < b.N; i++ {
    var x int
    n, err := fmt.Sscanf(numstr, "%d", &x)
    if n != 1 || x != num || err != nil {
      b.Error(err)
    }
  }
}

Você pode executá-lo salvando como atoi_test.goe executando go test -bench=. atoi_test.go.

goos: darwin
goarch: amd64
BenchmarkStrconvParseInt-8      100000000           17.1 ns/op
BenchmarkAtoi-8                 100000000           19.4 ns/op
BenchmarkFmtSscan-8               2000000          693   ns/op
PASS
ok      command-line-arguments  5.797s

2

Tente isto

import ("strconv")

value : = "123"
number,err := strconv.ParseUint(value, 10, 32)

0

Se você controlar os dados de entrada, poderá usar a mini versão

package main

import (
    "testing"
    "strconv"
)

func Atoi (s string) int {
    var (
        n uint64
        i int
        v byte
    )   
    for ; i < len(s); i++ {
        d := s[i]
        if '0' <= d && d <= '9' {
            v = d - '0'
        } else if 'a' <= d && d <= 'z' {
            v = d - 'a' + 10
        } else if 'A' <= d && d <= 'Z' {
            v = d - 'A' + 10
        } else {
            n = 0; break        
        }
        n *= uint64(10) 
        n += uint64(v)
    }
    return int(n)
}

func BenchmarkAtoi(b *testing.B) {
    for i := 0; i < b.N; i++ {
        in := Atoi("9999")
        _ = in
    }   
}

func BenchmarkStrconvAtoi(b *testing.B) {
    for i := 0; i < b.N; i++ {
        in, _ := strconv.Atoi("9999")
        _ = in
    }   
}

a opção mais rápida (faça seu cheque, se necessário). Resultado:

Path>go test -bench=. atoi_test.go
goos: windows
goarch: amd64
BenchmarkAtoi-2                 100000000               14.6 ns/op
BenchmarkStrconvAtoi-2          30000000                51.2 ns/op
PASS
ok      path     3.293s

1
O que ? Sério ? As pessoas que escreveram "ir" tornaram muito fácil. Não girar a roda :)
Balaji Boggaram Ramanarayan

E quanto a Atoi ("- 9999")?
Oleksiy Chechel
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.