当前位置：首页 > 文章列表 > Golang > Go教程 > 用bufio.Scanner快速读取整数文件到Go数组

用bufio.Scanner快速读取整数文件到Go数组

2025-09-19 16:07:07 0浏览收藏

想要高效读取整数文件到 Go 数组？本文为你介绍一种使用 `bufio.Scanner` 的方法，相较于 `fmt.Fscanf`，它拥有更简洁的错误处理机制，并且更符合 Go 语言的编程习惯。本文将深入讲解如何利用 `bufio.Scanner` 和 `io.Reader` 接口，从各种输入源（如文件、字符串等）中高效地读取整数数据。通过示例代码和详细解析，你将掌握如何使用 `scanner.Split(bufio.ScanWords)` 按单词分割输入，以及如何使用 `strconv.Atoi()` 将字符串转换为整数。文章还重点讨论了错误处理、性能优势和注意事项，助你编写出更健壮、更易于维护的 Go 代码，提升数据处理效率。

使用 bufio.Scanner 更高效地将整数文件读取到 Go 数组中

本文将介绍如何使用 bufio.Scanner 来高效地将包含整数的文件读取到 Go 语言的整数数组中。相比于使用 fmt.Fscanf，bufio.Scanner 提供了更简洁的错误处理方式，并且更加符合 Go 语言的编程习惯。此外，我们将使用 io.Reader 接口，使代码更加通用，可以处理任何实现了 io.Reader 接口的对象，而不仅仅是磁盘文件。

使用 bufio.Scanner 读取整数文件

bufio.Scanner 是 Go 标准库中用于读取文本的强大工具。它可以按行、按单词或使用自定义分割函数来扫描输入。在我们的例子中，我们将使用它按单词扫描，并将每个单词转换为整数。

以下是一个使用 bufio.Scanner 读取整数文件的示例代码：

package main

import (
    "bufio"
    "fmt"
    "io"
    "strconv"
    "strings"
)

// ReadInts 从 r 中读取以空格分隔的整数。如果发生错误，它将返回到目前为止成功读取的整数以及错误值。
func ReadInts(r io.Reader) ([]int, error) {
    scanner := bufio.NewScanner(r)
    scanner.Split(bufio.ScanWords)
    var result []int
    for scanner.Scan() {
        x, err := strconv.Atoi(scanner.Text())
        if err != nil {
            return result, err
        }
        result = append(result, x)
    }
    return result, scanner.Err()
}

func main() {
    tf := "1\n2\n3\n4\n5\n6"
    ints, err := ReadInts(strings.NewReader(tf))
    fmt.Println(ints, err)
}

代码解析：

ReadInts(r io.Reader) ([]int, error) 函数:
- 接受一个 io.Reader 接口作为输入，这意味着它可以从任何实现了 io.Reader 接口的对象读取数据，例如文件、字符串等。
- 创建一个 bufio.Scanner 对象，并将其与 io.Reader 关联。
- 使用 scanner.Split(bufio.ScanWords) 设置扫描器按单词分割输入。
- 循环调用 scanner.Scan() 读取下一个单词。
- 使用 strconv.Atoi() 将读取的单词转换为整数。如果转换失败，则返回已读取的整数数组和错误信息。
- 将转换后的整数添加到结果数组中。
- 循环结束后，返回结果数组和扫描器的错误信息（如果存在）。
main() 函数:
- 创建一个包含整数的字符串 tf，模拟文件内容。
- 使用 strings.NewReader(tf) 将字符串转换为 io.Reader 对象。
- 调用 ReadInts() 函数读取整数数组和错误信息。
- 打印结果数组和错误信息。