polars.LazyFrame.unnest#

LazyFrame.unnest(names: str | list[str]) LDF[source]#

Decompose a struct into its fields.

The fields will be inserted into the DataFrame on the location of the struct type.

Parameters:
names

Names of the struct columns that will be decomposed by its fields

Examples

>>> df = (
...     pl.DataFrame(
...         {
...             "before": ["foo", "bar"],
...             "t_a": [1, 2],
...             "t_b": ["a", "b"],
...             "t_c": [True, None],
...             "t_d": [[1, 2], [3]],
...             "after": ["baz", "womp"],
...         }
...     )
...     .lazy()
...     .select(
...         ["before", pl.struct(pl.col("^t_.$")).alias("t_struct"), "after"]
...     )
... )
>>> df.fetch()
shape: (2, 3)
┌────────┬─────────────────────┬───────┐
│ before ┆ t_struct            ┆ after │
│ ---    ┆ ---                 ┆ ---   │
│ str    ┆ struct[4]           ┆ str   │
╞════════╪═════════════════════╪═══════╡
│ foo    ┆ {1,"a",true,[1, 2]} ┆ baz   │
├╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
│ bar    ┆ {2,"b",null,[3]}    ┆ womp  │
└────────┴─────────────────────┴───────┘
>>> df.unnest("t_struct").fetch()
shape: (2, 6)
┌────────┬─────┬─────┬──────┬───────────┬───────┐
│ before ┆ t_a ┆ t_b ┆ t_c  ┆ t_d       ┆ after │
│ ---    ┆ --- ┆ --- ┆ ---  ┆ ---       ┆ ---   │
│ str    ┆ i64 ┆ str ┆ bool ┆ list[i64] ┆ str   │
╞════════╪═════╪═════╪══════╪═══════════╪═══════╡
│ foo    ┆ 1   ┆ a   ┆ true ┆ [1, 2]    ┆ baz   │
├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
│ bar    ┆ 2   ┆ b   ┆ null ┆ [3]       ┆ womp  │
└────────┴─────┴─────┴──────┴───────────┴───────┘