CODEWITHSUNDEEP
arraysgoslice
Jonas D
Jonas D

Reputation: 298

selecting 2D sub-slice of a 2D-slice using ranges in go

I'm getting a surprising result when selecting a 2D sub-slice of a slice. Consider the following 2D int array

a := [][]int{
    {0, 1, 2, 3},
    {1, 2, 3, 4},
    {2, 3, 4, 5},
    {3, 4, 5, 6},
}

To select the top left 3x3 2D slice using ranges I would use

b := a[0:2][0:2]

I would expect the result to be

[[0 1 2] [1 2 3] [2 3 4]]

however the second index range doesn't seem to have any effect, and returns the following instead:

[[0 1 2 3] [1 2 3 4] [2 3 4 5]]

What am I missing? Can you simply not select a sub-slice like this where the dimension > 1 ?

Upvotes: 6

Views: 2082

Answers (1)

icza
icza

Reputation: 418037

You can't do what you want in a single step. Slices and arrays are not 2-dimensional, they are just composed to form a multi-dimensional object. See How is two dimensional array's memory representation

So with a slice expression, you just get a slice that will hold a subset of the "full" rows, and its type will be the same: [][]int. If you slice it again, you just slicing the slice of rows again.

Also note that the higher index in a slice expression is exclusive, so a[0:2] will only have 2 rows, so you should use a[0:3] or simply a[:3] instead.

To get what you want, you have to slice the rows individually like this:

b := a[0:3]
for i := range b {
    b[i] = b[i][0:3]
}
fmt.Println(b)

This will output (try it on the Go Playground):

[[0 1 2] [1 2 3] [2 3 4]]

Or shorter:

b := a[:3]
for i, bi := range b {
    b[i] = bi[:3]
}

Upvotes: 7

Related Questions