63 lines
895 B
Go
63 lines
895 B
Go
|
package clusters
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"encoding/csv"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
type csvImporter struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func CsvImporter() Importer {
|
||
|
|
return &csvImporter{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *csvImporter) Import(file string, start, end int) ([][]float64, error) {
|
||
|
|
if start < 0 || end < 0 || start > end {
|
||
|
|
return [][]float64{}, errInvalidRange
|
||
|
|
}
|
||
|
|
|
||
|
|
f, err := os.Open(file)
|
||
|
|
if err != nil {
|
||
|
|
return [][]float64{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
defer f.Close()
|
||
|
|
|
||
|
|
var (
|
||
|
|
d = make([][]float64, 0)
|
||
|
|
r = csv.NewReader(bufio.NewReader(f))
|
||
|
|
s = end - start + 1
|
||
|
|
g []float64
|
||
|
|
)
|
||
|
|
|
||
|
|
Main:
|
||
|
|
for {
|
||
|
|
record, err := r.Read()
|
||
|
|
|
||
|
|
if err == io.EOF {
|
||
|
|
break
|
||
|
|
} else if err != nil {
|
||
|
|
return [][]float64{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
g = make([]float64, 0, s)
|
||
|
|
|
||
|
|
for j := start; j <= end; j++ {
|
||
|
|
f, err := strconv.ParseFloat(record[j], 64)
|
||
|
|
if err == nil {
|
||
|
|
g = append(g, f)
|
||
|
|
} else {
|
||
|
|
continue Main
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
d = append(d, g)
|
||
|
|
}
|
||
|
|
|
||
|
|
return d, nil
|
||
|
|
}
|