将 os.Stdin 中的数字读入 Go 中的数组或切片

Posted 2023-02-23

【中文标题】将 os.Stdin 中的数字读入 Go 中的数组或切片

【英文标题】：Read numbers from os.Stdin into array or slice in Go

【发布时间】：2018-08-03 05:36:56

【问题描述】：

我可以使用下面的多个变量读取多个数字。

numbers := make([]int, 2)
fmt.Fscan(os.Stdin, &numbers[0], &numbers[1])

我可以使用数字本身而不是单独的形状来使其更简单吗？

numbers := make([]int, 2)
fmt.Fscan(os.Stdin, &numbers) // just an example, got error

【参考方案1】：

fmt 包不处理扫描切片，但您可以创建一个实用函数来打包所有元素的地址：

func packAddrs(n []int) []interface 
    p := make([]interface, len(n))
    for i := range n 
        p[i] = &n[i]
    
    return p

使用它，您可以像这样扫描整个切片：

numbers := make([]int, 2)
n, err := fmt.Fscan(os.Stdin, packAddrs(numbers)...)
fmt.Println(numbers, n, err)

用fmt.Sscan()测试它：

numbers := make([]int, 5)
n, err := fmt.Sscan("1 3 5 7 9", packAddrs(numbers)...)
fmt.Println(numbers, n, err)

输出（在Go Playground 上试试）：

[1 3 5 7 9] 5 <nil>