TC_func_stdlib_get_dataframe.go

package ThreadComputation

import (
  log "github.com/sirupsen/logrus"
)

func TCDataframeGet(x interface{}, y interface{}) interface{} {
  out := new(TCList)

  switch data :=  x.(type) {
  case *TCData:
    switch y.(type) {
    case string:
      c, err := data.D.NameToColumn(y.(string))
      if err != nil {
        log.Errorf("DataFrame error: %v", err)
        return nil
      }
      iterator := data.D.Series[c].ValuesIterator()
      for  {
        n, d, _ := iterator()
        if n == nil {
          break
        }
        out.Q.PushBack(d)
      }
    }
  }
  return out
}

func init() {
  RegisterOperatorCallback("get", Data, String, TCDataframeGet)
}