Reshape this Expr to a flat Series or a Series of Lists

Description

Usage

<Expr>$reshape(dimensions)

Arguments

dimensions A integer vector of length of the dimension size. If -1 is used in any of the dimensions, that dimension is inferred. Currently, more than two dimensions not supported.

Value

Expr. If a single dimension is given, results in an expression of the original data type. If a multiple dimensions are given, results in an expression of data type List with shape equal to the dimensions.

Examples

library("polars")

df = pl$DataFrame(foo = 1:9)

df$select(pl$col("foo")$reshape(9))

#> shape: (9, 1)
#> ┌─────┐
#> │ foo │
#> │ --- │
#> │ i32 │
#> ╞═════╡
#> │ 1   │
#> │ 2   │
#> │ 3   │
#> │ 4   │
#> │ 5   │
#> │ 6   │
#> │ 7   │
#> │ 8   │
#> │ 9   │
#> └─────┘

df$select(pl$col("foo")$reshape(c(3, 3)))

#> shape: (3, 1)
#> ┌───────────────┐
#> │ foo           │
#> │ ---           │
#> │ array[i32, 3] │
#> ╞═══════════════╡
#> │ [1, 2, 3]     │
#> │ [4, 5, 6]     │
#> │ [7, 8, 9]     │
#> └───────────────┘

# Use `-1` to infer the other dimension
df$select(pl$col("foo")$reshape(c(-1, 3)))

#> shape: (3, 1)
#> ┌───────────────┐
#> │ foo           │
#> │ ---           │
#> │ array[i32, 3] │
#> ╞═══════════════╡
#> │ [1, 2, 3]     │
#> │ [4, 5, 6]     │
#> │ [7, 8, 9]     │
#> └───────────────┘

df$select(pl$col("foo")$reshape(c(3, -1)))

#> shape: (3, 1)
#> ┌───────────────┐
#> │ foo           │
#> │ ---           │
#> │ array[i32, 3] │
#> ╞═══════════════╡
#> │ [1, 2, 3]     │
#> │ [4, 5, 6]     │
#> │ [7, 8, 9]     │
#> └───────────────┘

# One can specify more than 2 dimensions by using the Array type
df = pl$DataFrame(foo = 1:12)
df$select(
  pl$col("foo")$reshape(c(3, 2, 2))
)

#> shape: (3, 1)
#> ┌─────────────────────┐
#> │ foo                 │
#> │ ---                 │
#> │ array[i32, (2, 2)]  │
#> ╞═════════════════════╡
#> │ [[1, 2], [3, 4]]    │
#> │ [[5, 6], [7, 8]]    │
#> │ [[9, 10], [11, 12]] │
#> └─────────────────────┘