如何获得我的数据的特定形状

时间:2018-07-19 06:32:16

标签: go slice shape

在Go中,我正在使用此功能bars, err := custplotter.NewCandlesticks(data) 从这里: https://github.com/pplcc/plotext/tree/master/custplotter

期望这种形状的数据:

[{2 16435 16458 16435 16446 1} {3 16446 16458 16435.04 16455 1} .....]

但是我下面的代码改为以这种形状创建数据:

[[2 16435 16458 16435 16446 1] [3 16446 16458 16435.04 16455 1] .....]

哪个给我这个错误信息: 无法在type [ ][ ]string的参数中将数据(custplotter.TOHLCVer)用作类型custplotter.NewCandlesticks

[ ][ ]string does not implement custplotter.TOHLCVer (missing Len method)

我认为问题在于数据形状。如何更改代码以创建所需的数据形状(用{}代替[])?

   //read excel file******************************************
    xlsx, err := excelize.OpenFile("/media/Snaps/test snaps.xlsm")
    if err != nil {
        fmt.Println(err)
        return
    }

    //read all rows into df
    df := xlsx.GetRows("ticker_2")

    //get only TOHLCV columns and 60 rows
    df3 := make([][]string, 60) // create slice for 60 rows
    idx := 0
    for _, row := range df[1:61] { // read 60 rows
        df3row := make([]string, 6) // create slice for 6 columns
        copy(df3row, row[28:34]) // copy desired columns to new row slice
        df3[idx] = df3row
        idx++
    }

我在Go文学中发现的所有切片示例仅使用[[],[]]

1 个答案:

答案 0 :(得分:0)

按照https://github.com/pplcc/plotext/blob/68ab3c6e05c34baf5af21c9f5c3341f527a110ac/examples/tohlcvexampledata.go#L42

似乎您需要的是custplotter.TOHLCVs,它只是float64结构的一部分。

https://github.com/pplcc/plotext/blob/master/custplotter/tohlcv.go

type TOHLCVer interface {
    // Len returns the number of time, open, high, low, close, volume tuples.
    Len() int

    // TOHLCV returns an time, open, high, low, close, volume tuple.
    TOHLCV(int) (float64, float64, float64, float64, float64, float64)
}

// TOHLCVs implements the TOHLCVer interface using a slice.
type TOHLCVs []struct{ T, O, H, L, C, V float64 }

所以基本上您的解决方案可能类似于:

df3 := make(TOHLCVs, 60) // create slice for 60 rows
idx := 0
for _, row := range df[1:61] { // read 60 rows
    df3[idx].T, err = strconv.ParseFloat(row[28], 64)
    df3[idx].O, err = strconv.ParseFloat(row[29], 64)
    df3[idx].H, err = strconv.ParseFloat(row[30], 64)
    df3[idx].L, err = strconv.ParseFloat(row[31], 64)
    df3[idx].C, err = strconv.ParseFloat(row[32], 64)
    df3[idx].V, err = strconv.ParseFloat(row[33], 64)
    idx++
}

或者您也可以实现TOHLCVer接口:)

type SlicesOfTOHLCV [][6]float64

func (s SlicesOfTOHLCV) Len() int {
    return len(s)
}

func (s SlicesOfTOHLCV) TOHLCV(i int) (float64, float64, float64, float64, float64) {
    return s[i][0], s[i][1], s[i][2], s[i][3], s[i][4], s[i][5]
}

mySlice := make(SlicesOfTOHLCV, 60)
i := 0
for _, row := range df[1:61] {
    mySlice[i] = [6]float64{}
    for j := 0; j < 6; j ++ {
        mySlice[i][j], err = strconv.ParseFloat(row[28+j], 64)
        if err != nil {
            panic(err)
        }
    }
    i ++
}