dongpixi2648 2017-09-10 18:10
浏览 65
已采纳

将来自控制台的字符串输入片段转换为数字片段

I'm trying to write a Go script that takes in as many lines of comma-separated coordinates as the user wishes, split and convert the string of coordinates to float64, store each line as a slice, and then append each slice in a slice of slices for later usage.

Example inputs are:

1.1,2.2,3.3
3.14,0,5.16

Example outputs are:

[[1.1 2.2 3.3],[3.14 0 5.16]]

The equivalent in Python is

def get_input():
    print("Please enter comma separated coordinates:")
    lines = []
    while True:
        line = input()
        if line:
            line = [float(x) for x in line.replace(" ", "").split(",")]
            lines.append(line)
        else:
            break
    return lines

But what I wrote in Go seems way too long (pasted below), and I'm creating a lot of variables without the ability to change variable type as in Python. Since I literally just started writing Golang to learn it, I fear my script is long as I'm trying to convert Python thinking into Go. Therefore, I would like to ask for some advice as to how to write this script shorter and more concise in Go style? Thank you.

package main

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

func main() {
    inputs := get_input()
    fmt.Println(inputs)
}

func get_input() [][]float64 {
    fmt.Println("Please enter comma separated coordinates: ")

    var inputs [][]float64

    scanner := bufio.NewScanner(os.Stdin)
    for scanner.Scan() {
        if len(scanner.Text()) > 0 {
            raw_input := strings.Replace(scanner.Text(), " ", "", -1)
            input := strings.Split(raw_input, ",")
            converted_input := str2float(input)
            inputs = append(inputs, converted_input)
        } else {
            break
        }
    }

    return inputs
}

func str2float(records []string) []float64 {

    var float_slice []float64

    for _, v := range records {
        if s, err := strconv.ParseFloat(v, 64); err == nil {
            float_slice = append(float_slice, s)
        }
    }

    return float_slice
}
  • 写回答

2条回答

  • doudao1282 2017-09-11 13:33
    关注

    Using only string functions:

    package main
    
    import (
        "bufio"
        "fmt"
        "os"
        "strconv"
        "strings"
    )
    
    func main() {
        scanner := bufio.NewScanner(os.Stdin)
        var result [][]float64
        var txt string
        for scanner.Scan() {
            txt = scanner.Text()
            if len(txt) > 0 {
                values := strings.Split(txt, ",")
                var row []float64
                for _, v := range values {
                    fl, err := strconv.ParseFloat(strings.Trim(v, " "), 64)
                    if err != nil {
                        panic(fmt.Sprintf("Incorrect value for float64 '%v'", v))
                    }
                    row = append(row, fl)
                }
                result = append(result, row)
            }
        }
        fmt.Printf("Result: %v
    ", result)
    }
    

    Run:

    $ printf "1.1,2.2,3.3
    3.14,0,5.16
    2,45,76.0, 45 , 69" | go run experiment2.go
    
    Result: [[1.1 2.2 3.3] [3.14 0 5.16] [2 45 76 45 69]]
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥60 版本过低apk如何修改可以兼容新的安卓系统
  • ¥25 由IPR导致的DRIVER_POWER_STATE_FAILURE蓝屏
  • ¥50 有数据,怎么建立模型求影响全要素生产率的因素
  • ¥50 有数据,怎么用matlab求全要素生产率
  • ¥15 TI的insta-spin例程
  • ¥15 完成下列问题完成下列问题
  • ¥15 C#算法问题, 不知道怎么处理这个数据的转换
  • ¥15 YoloV5 第三方库的版本对照问题
  • ¥15 请完成下列相关问题!
  • ¥15 drone 推送镜像时候 purge: true 推送完毕后没有删除对应的镜像,手动拷贝到服务器执行结果正确在样才能让指令自动执行成功删除对应镜像,如何解决?