Skip to content

Instantly share code, notes, and snippets.

@hankliu5
Created April 2, 2018 02:20
Show Gist options
  • Save hankliu5/6b814f479683de907e082a4d7fc9ebca to your computer and use it in GitHub Desktop.
Save hankliu5/6b814f479683de907e082a4d7fc9ebca to your computer and use it in GitHub Desktop.
package main
import (
"fmt"
"strings"
"os"
"bufio"
"io"
"strconv"
"gonum.org/v1/gonum/mat"
)
func main() {
if len(os.Args) < 2 {
fmt.Printf("Usage: %s <filename>", os.Args[0])
}
file, err := os.Open(os.Args[1])
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer file.Close()
var num_list []float64
row := 0
num_elem := 0
r := bufio.NewReader(file)
for {
s, e := r.ReadString('\n')
if e != nil {
if e != io.EOF {
fmt.Println(e)
}
break
}
row += 1
slist := strings.Split(s[:len(s)-1], " ")
for _, element := range slist {
num, _ := strconv.ParseFloat(element, 64)
num_list = append(num_list, num)
num_elem += 1
}
}
a := mat.NewDense(row, num_elem/row, num_list)
fa := mat.Formatted(a, mat.Prefix(" "), mat.Squeeze())
fmt.Printf("c = %v", fa)
fmt.Printf("\n")
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment