map[0:DataFrame error: error on series 0: indexing error: unknown indexing mode 1:DataFrame error: error on series 0: indexing error: unknown indexing mode]
package main
import (
"fmt"
"log"
"os"
"github.com/go-gota/gota/dataframe"
)
func main() {
fileOpen, err := os.Open("./advertising.csv")
if err != nil {
log.Fatal(err)
}
CSVRead := dataframe.ReadCSV(fileOpen)
trainData := CSVRead.Nrow() / 5
testData := (CSVRead.Nrow() * 4) / 5
if trainData+testData < CSVRead.Nrow() {
trainData++
}
trainDf := make([]float64, trainData)
testDf := make([]float64, testData)
for i := 0; i < trainData; i++ {
trainDf[i] = float64(i)
}
for i := 0; i < testData; i++ {
testDf[i] = float64(trainData) + float64(i)
}
dfTrain := CSVRead.Subset(trainDf)
dfTest := CSVRead.Subset(testDf)
// CSVName := []string{"train.csv", "test.csv"}
setMap := map[int]dataframe.DataFrame{
0: dfTrain,
1: dfTest,
}
fmt.Println(setMap)
// for i, setName := range CSVName {
// f, err := os.Create(setName)
// if err != nil {
// log.Fatal(err)
// }
// writer := bufio.NewWriter(f)
// if err := setMap[float64(i)].WriteCSV(writer); err != nil {
// fmt.Println(err)
// }
// }
}