dynamic array dimension in golang

Question

I am trying to create a high dimension array in Golang.
Does anyone know how to do it?

e.g.

• dims := [3,2,1] as a parameter -> want high_dims_array := make([3][2][1]int, 0)
• dims := [2] -> want high_dims_array := make([2]int, 0)
• dims := [3,3] -> want high_dims_array := make([3][3]int, 0)

Where the dims is a variable containing dimensions.

Answer 1

Thanks my friends. I have figured out a way to do this

func initialCube(shape []int) []interface{} {
    // base condition
    if len(shape) <= 1 {
        dim := shape[len(shape)-1]
        retObj := make([]interface{}, dim)
        for i := 0; i < dim; i++ {
            retObj[i] = 0.0
        }
        return retObj
    } else { // recursive
        dim := shape[len(shape)-1]
        retObj := make([]interface{}, dim)
        for i := 0; i < dim; i++ {
            retObj[i] = initialCube(shape[:len(shape)-1])
        }
        return retObj
    }
}

Answer 2

That looks like what dolmen-go/multidim does (it helps to allocate a slice with the required number of elements):

package main

import (
    "fmt"

    "github.com/dolmen-go/multidim"
)

func main() {
    var cube [][][]int
    multidim.Init(&cube, 8, 2, 2, 2)

    fmt.Println(cube)
}

Output:

[[[8 8] [8 8]] [[8 8] [8 8]]]

You can also use a function (still with the same library) to initialize your 3*2 slice:

package main

import (
    "fmt"

    "github.com/dolmen-go/multidim"
)

func main() {
    var a [][]int

    multidim.Init(&a, func(i, j int) int {
        return 2*i + j + 1
    }, 3, 2)

    fmt.Println(a)

    var r [][]string

    multidim.Init(&r, func(i, j int) string {
        return "foobar"[i*3+j : i*3+j+1]
    }, 2, 3)

    fmt.Println(r)
}

Output:

[[1 2] [3 4] [5 6]]
[[f o o] [b a r]]