Deion Pandas has support for "Make a MultiIndex from the car

Might be duplicate of <a class="issue-link js-issue-link" data-error-text="Failed to l

Support for making all possible combinations of elements in a df, multiIndex from the cartesian product of multiple iterables? about polars HOT 2 OPEN

JacobElder commented on July 21, 2024

Support for making all possible combinations of elements in a df, multiIndex from the cartesian product of multiple iterables?

from polars.

Comments (2)

etiennebacher commented on July 21, 2024

Might be duplicate of #9722

from polars.

mcrumiller commented on July 21, 2024

Here is a function do that with basic Series, it's easy to extend to other iterables.

from datetime import date
import polars as pl


def cross(series_list):
    """Generate a DataFrame consisting of all combinations of values in a list of Series."""
    lfs = [pl.LazyFrame(s) for s in series_list]
    out = lfs[0]
    for lf in lfs[1:]:
        out = out.join(lf, how="cross")
    return out.collect()


s1 = pl.Series("int", [1, 2, 3])
s2 = pl.Series("char", ["a", "b", "c", "d"])
s3 = pl.Series("date", [date(2024, 1, 1), date(2024, 1, 2)])

with pl.Config(tbl_rows=24):
    print(cross([s1, s2, s3]))

shape: (24, 3)
┌─────┬──────┬────────────┐
│ int ┆ char ┆ date       │
│ --- ┆ ---  ┆ ---        │
│ i64 ┆ str  ┆ date       │
╞═════╪══════╪════════════╡
│ 1   ┆ a    ┆ 2024-01-01 │
│ 1   ┆ a    ┆ 2024-01-02 │
│ 1   ┆ b    ┆ 2024-01-01 │
│ 1   ┆ b    ┆ 2024-01-02 │
│ 1   ┆ c    ┆ 2024-01-01 │
│ 1   ┆ c    ┆ 2024-01-02 │
│ 1   ┆ d    ┆ 2024-01-01 │
│ 1   ┆ d    ┆ 2024-01-02 │
│ 2   ┆ a    ┆ 2024-01-01 │
│ 2   ┆ a    ┆ 2024-01-02 │
│ 2   ┆ b    ┆ 2024-01-01 │
│ 2   ┆ b    ┆ 2024-01-02 │
│ 2   ┆ c    ┆ 2024-01-01 │
│ 2   ┆ c    ┆ 2024-01-02 │
│ 2   ┆ d    ┆ 2024-01-01 │
│ 2   ┆ d    ┆ 2024-01-02 │
│ 3   ┆ a    ┆ 2024-01-01 │
│ 3   ┆ a    ┆ 2024-01-02 │
│ 3   ┆ b    ┆ 2024-01-01 │
│ 3   ┆ b    ┆ 2024-01-02 │
│ 3   ┆ c    ┆ 2024-01-01 │
│ 3   ┆ c    ┆ 2024-01-02 │
│ 3   ┆ d    ┆ 2024-01-01 │
│ 3   ┆ d    ┆ 2024-01-02 │
└─────┴──────┴────────────┘

from polars.

Recommend Projects