Início / coding / Perguntas / 79188007
Accepted
Andi
Asked: 2024-11-14 17:11:39 +0800 CST

Equivalente polar de numpy.tile

  • 772
df = pl. DataFrame({"col1": [1, 2, 3], "col2": [4, 5, 6]})

print(df)

shape: (3, 2)
┌──────┬──────┐
│ col1 ┆ col2 │
│ ---  ┆ ---  │
│ i64  ┆ i64  │
╞══════╪══════╡
│ 1    ┆ 4    │
│ 2    ┆ 5    │
│ 3    ┆ 6    │
└──────┴──────┘

Estou procurando o polarsequivalente a numpy.tile.
Algo parecido com df.tile(2)ou df.select(pl.all().tile(2)).

O resultado esperado deve ser semelhante a este:

shape: (6, 2)
┌──────┬──────┐
│ col1 ┆ col2 │
│ ---  ┆ ---  │
│ i64  ┆ i64  │
╞══════╪══════╡
│ 1    ┆ 4    │
│ 2    ┆ 5    │
│ 3    ┆ 6    │
│ 1    ┆ 4    │
│ 2    ┆ 5    │
│ 3    ┆ 6    │
└──────┴──────┘
python

1 respostas

  1. Best Answer

    Você pode repetir concatvárias vezes o mesmo DataFrame:

    out = pl.concat([df]*2)

# for contiguous memory
out = pl.concat([df]*2, rechunk=True)

    Ou, para apenas 2 repetições vstack:

    out = df.vstack(df)

    Alternativamente, usando Expr.repeat_by+ explode:

    N = 2

(df.with_columns(pl.all().repeat_by(pl.lit(N)), pl.int_ranges(pl.lit(N)))
   .explode(pl.all())
   .sort(by='literal', maintain_order=True).drop('literal')
)

    Saída:

    ┌──────┬──────┐
│ col1 ┆ col2 │
│ ---  ┆ ---  │
│ i64  ┆ i64  │
╞══════╪══════╡
│ 1    ┆ 4    │
│ 2    ┆ 5    │
│ 3    ┆ 6    │
│ 1    ┆ 4    │
│ 2    ┆ 5    │
│ 3    ┆ 6    │
└──────┴──────┘
    • 1

relate perguntas