在 Go 中读取一行数字

Question

我有以下输入，其中第一行是 N - 数字计数，第二行是 N 个数字，以空格分隔：5 2 1 0 3 4在 Python 中，我可以在不指定计数 (N) 的情况下读取数字：_ = input() numbers = list(map(int, input().split()))我怎样才能在 Go 中做同样的事情？或者我必须确切地知道有多少个数字？

Answer 1

您可以使用 bufio 逐行遍历文件，并且该strings模块可以将字符串拆分为 slice。所以这让我们得到类似的东西：

package main

import (

    "bufio"

    "fmt"

    "os"

    "strconv"

    "strings"

)

func main() {

    readFile, err := os.Open("data.txt")

    defer readFile.Close()

    if err != nil {

        fmt.Println(err)

    }

    fileScanner := bufio.NewScanner(readFile)

    fileScanner.Split(bufio.ScanLines)

    for fileScanner.Scan() {

        // get next line from the file

        line := fileScanner.Text()

        // split it into a list of space-delimited tokens

        chars := strings.Split(line, " ")

        // create an slice of ints the same length as

        // the chars slice

        ints := make([]int, len(chars))

        for i, s := range chars {

            // convert string to int

            val, err := strconv.Atoi(s)

            if err != nil {

                panic(err)

            }

            // update the corresponding position in the

            // ints slice

            ints[i] = val

        }

        fmt.Printf("%v\n", ints)

    }

}

给定您的样本数据将输出：

[5]

[2 1 0 3 4]

Answer 2

由于您知道定界符并且只有 2 行，因此这也是一个更紧凑的解决方案：

package main

import (

    "fmt"

    "os"

    "regexp"

    "strconv"

    "strings"

)

func main() {

    parts, err := readRaw("data.txt")

    if err != nil {

        panic(err)

    }

    n, nums, err := toNumbers(parts)

    if err != nil {

        panic(err)

    }

    fmt.Printf("%d: %v\n", n, nums)

}

// readRaw reads the file in input and returns the numbers inside as a slice of strings

func readRaw(fn string) ([]string, error) {

    b, err := os.ReadFile(fn)

    if err != nil {

        return nil, err

    }

    return regexp.MustCompile(`\s`).Split(strings.TrimSpace(string(b)), -1), nil

}

// toNumbers plays with the input string to return the data as a slice of int

func toNumbers(parts []string) (int, []int, error) {

    n, err := strconv.Atoi(parts[0])

    if err != nil {

        return 0, nil, err

    }

    nums := make([]int, 0)

    for _, p := range parts[1:] {

        num, err := strconv.Atoi(p)

        if err != nil {

            return n, nums, err

        }

        nums = append(nums, num)

    }

    return n, nums, nil

}

输出是：

5: [2 1 0 3 4]