将从控制台输入的字符串切片转换为数字切片。

huangapple go评论147阅读模式
英文:

Convert slice of string input from console to slice of numbers

问题

我正在尝试编写一个Go脚本,它接受用户希望输入的任意行以逗号分隔的坐标,将坐标字符串拆分并转换为float64,将每行存储为切片,然后将每个切片附加到切片的切片中以供以后使用。

示例输入为:

1.1,2.2,3.3
3.14,0,5.16

示例输出为:

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

在Python中的等效代码如下:

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

但是我在Go中编写的代码似乎太长了(如下所示),而且我创建了很多变量,而且不能像Python那样更改变量类型。由于我刚刚开始学习Go语言,我担心我的脚本太长了,因为我试图将Python的思维转换为Go。因此,我想请教一些建议,如何以更短、更简洁的Go风格编写这个脚本?谢谢。

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
}
英文:

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
}

答案1

得分: 2

使用字符串函数:

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\n", result)
}

运行:

$ 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]]

以上是给定的代码,它使用字符串函数将输入的文本解析为浮点数矩阵,并将结果打印出来。

英文:

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\n", 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]]

答案2

得分: 1

使用给定的输入,您可以将它们连接起来以生成一个JSON字符串,然后进行反序列化:

func main() {
    var lines []string
    for {
        var line string
        fmt.Scanln(&line)
        if line == "" {
            break
        }
        lines = append(lines, "["+line+"]")
    }
    all := "[" + strings.Join(lines, ",") + "]"
    inputs := [][]float64{}
    if err := json.Unmarshal([]byte(all), &inputs); err != nil {
        fmt.Println(err)
        return
    }

    fmt.Println(inputs)
}
英文:

With given input, you can concatenate them to make a JSON string and then unmarshal (deserialize) that:

func main() {
	var lines []string
	for {
		var line string
		fmt.Scanln(&line)
		if line == "" {
			break
		}
		lines = append(lines, "["+line+"]")
	}
	all := "[" + strings.Join(lines, ",") + "]"
	inputs := [][]float64{}
	if err := json.Unmarshal([]byte(all), &inputs); err != nil {
		fmt.Println(err)
		return
	}

	fmt.Println(inputs)
}

huangapple
  • 本文由 发表于 2017年9月11日 02:10:09
  • 转载请务必保留本文链接:https://go.coder-hub.com/46143974.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定