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

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

Convert slice of string input from console to slice of numbers

问题

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

示例输入为：

  1. 1.1,2.2,3.3
  2. 3.14,0,5.16

示例输出为：

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

在Python中的等效代码如下：

  1. def get_input():
  2.     print("Please enter comma separated coordinates:")
  3.     lines = []
  4.     while True:
  5.         line = input()
  6.         if line:
  7.             line = [float(x) for x in line.replace(" ", "").split(",")]
  8.             lines.append(line)
  9.         else:
  10.             break
  11.     return lines

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

  1. package main
  3. import (
  4.     "fmt"
  5.     "os"
  6.     "bufio"
  7.     "strings"
  8.     "strconv"
  9. )
  11. func main() {
  12.     inputs := get_input()
  13.     fmt.Println(inputs)
  14. }
  16. func get_input() [][]float64 {
  17.     fmt.Println("Please enter comma separated coordinates: ")
  19.     var inputs [][]float64
  21.     scanner := bufio.NewScanner(os.Stdin)
  22.     for scanner.Scan() {
  23.         if len(scanner.Text()) > 0 {
  24.             raw_input := strings.Replace(scanner.Text(), " ", "", -1)
  25.             input := strings.Split(raw_input, ",")
  26.             converted_input := str2float(input)
  27.             inputs = append(inputs, converted_input)
  28.         } else {
  29.             break
  30.         }
  31.     }
  33.     return inputs
  34. }
  36. func str2float(records []string) []float64 {
  37.     
  38.     var float_slice []float64
  39.     
  40.     for _, v := range records {
  41.         if s, err := strconv.ParseFloat(v, 64); err == nil {
  42.             float_slice = append(float_slice, s)
  43.         }
  44.     }
  46.     return float_slice
  47. }
英文:

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.1,2.2,3.3
  2. 3.14,0,5.16

Example outputs are:

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

The equivalent in Python is

  1. def get_input():
  2.     print("Please enter comma separated coordinates:")
  3.     lines = []
  4.     while True:
  5.         line = input()
  6.         if line:
  7.             line = [float(x) for x in line.replace(" ", "").split(",")]
  8.             lines.append(line)
  9.         else:
  10.             break
  11.     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.

  1. package main
  3. import (
  4.     "fmt"
  5.     "os"
  6.     "bufio"
  7.     "strings"
  8.     "strconv"
  9. )
  11. func main() {
  12.     inputs := get_input()
  13.     fmt.Println(inputs)
  14. }
  16. func get_input() [][]float64 {
  17.     fmt.Println("Please enter comma separated coordinates: ")
  19.     var inputs [][]float64
  21.     scanner := bufio.NewScanner(os.Stdin)
  22.     for scanner.Scan() {
  23.         if len(scanner.Text()) > 0 {
  24.             raw_input := strings.Replace(scanner.Text(), " ", "", -1)
  25.             input := strings.Split(raw_input, ",")
  26.             converted_input := str2float(input)
  27.             inputs = append(inputs, converted_input)
  28.         } else {
  29.             break
  30.         }
  31.     }
  33.     return inputs
  34. }
  36. func str2float(records []string) []float64 {
  37.     
  38.     var float_slice []float64
  39.     
  40.     for _, v := range records {
  41.         if s, err := strconv.ParseFloat(v, 64); err == nil {
  42.             float_slice = append(float_slice, s)
  43.         }
  44.     }
  46.     return float_slice
  47. }

答案1

得分: 2

使用字符串函数：

  1. package main
  3. import (
  4. 	"bufio"
  5. 	"fmt"
  6. 	"os"
  7. 	"strconv"
  8. 	"strings"
  9. )
  11. func main() {
  12. 	scanner := bufio.NewScanner(os.Stdin)
  13. 	var result [][]float64
  14. 	var txt string
  15. 	for scanner.Scan() {
  16. 		txt = scanner.Text()
  17. 		if len(txt) > 0 {
  18. 			values := strings.Split(txt, ",")
  19. 			var row []float64
  20. 			for _, v := range values {
  21. 				fl, err := strconv.ParseFloat(strings.Trim(v, " "), 64)
  22. 				if err != nil {
  23. 					panic(fmt.Sprintf("Incorrect value for float64 '%v'", v))
  24. 				}
  25. 				row = append(row, fl)
  26. 			}
  27. 			result = append(result, row)
  28. 		}
  29. 	}
  30. 	fmt.Printf("Result: %v\n", result)
  31. }

运行：

  1. $ printf "1.1,2.2,3.3
  2. 3.14,0,5.16
  3. 2,45,76.0, 45 , 69" | go run experiment2.go
  5. Result: [[1.1 2.2 3.3] [3.14 0 5.16] [2 45 76 45 69]]

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

英文:

Using only string functions:

  1. package main
  3. import (
  4. 	"bufio"
  5. 	"fmt"
  6. 	"os"
  7. 	"strconv"
  8. 	"strings"
  9. )
  11. func main() {
  12. 	scanner := bufio.NewScanner(os.Stdin)
  13. 	var result [][]float64
  14. 	var txt string
  15. 	for scanner.Scan() {
  16. 		txt = scanner.Text()
  17. 		if len(txt) > 0 {
  18. 			values := strings.Split(txt, ",")
  19. 			var row []float64
  20. 			for _, v := range values {
  21. 				fl, err := strconv.ParseFloat(strings.Trim(v, " "), 64)
  22. 				if err != nil {
  23. 					panic(fmt.Sprintf("Incorrect value for float64 '%v'", v))
  24. 				}
  25. 				row = append(row, fl)
  26. 			}
  27. 			result = append(result, row)
  28. 		}
  29. 	}
  30. 	fmt.Printf("Result: %v\n", result)
  31. }

Run:

  1. $ printf "1.1,2.2,3.3
  2. 3.14,0,5.16
  3. 2,45,76.0, 45 , 69" | go run experiment2.go
  5. Result: [[1.1 2.2 3.3] [3.14 0 5.16] [2 45 76 45 69]]

答案2

得分: 1

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

  1. func main() {
  2.     var lines []string
  3.     for {
  4.         var line string
  5.         fmt.Scanln(&line)
  6.         if line == "" {
  7.             break
  8.         }
  9.         lines = append(lines, "["+line+"]")
  10.     }
  11.     all := "[" + strings.Join(lines, ",") + "]"
  12.     inputs := [][]float64{}
  13.     if err := json.Unmarshal([]byte(all), &inputs); err != nil {
  14.         fmt.Println(err)
  15.         return
  16.     }
  18.     fmt.Println(inputs)
  19. }
英文:

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

  1. func main() {
  2. 	var lines []string
  3. 	for {
  4. 		var line string
  5. 		fmt.Scanln(&line)
  6. 		if line == "" {
  7. 			break
  8. 		}
  9. 		lines = append(lines, "["+line+"]")
  10. 	}
  11. 	all := "[" + strings.Join(lines, ",") + "]"
  12. 	inputs := [][]float64{}
  13. 	if err := json.Unmarshal([]byte(all), &inputs); err != nil {
  14. 		fmt.Println(err)
  15. 		return
  16. 	}
  18. 	fmt.Println(inputs)
  19. }

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

发表评论

匿名网友

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

确定