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

时间:2018-02-23 07:34:17

标签: go

我可以使用bellow等多个变量读取多个数字。

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 个答案:

答案 0 :(得分:2)

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>