主页 / coding / 问题 / 79188007
Accepted
Andi
Asked: 2024-11-14 17:11:39 +0800 CST

与 numpy.tile 相当的 Polars

  • 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    │
└──────┴──────┘

我正在寻找polars相当于 的东西numpy.tile。
类似于df.tile(2)或 的东西df.select(pl.all().tile(2))。

预期结果应如下所示：

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

1 个回答

  1. Best Answer

    你可以concat多次使用相同的DataFrame：

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

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

    或者，仅重复 2 次vstack：

    out = df.vstack(df)

    或者，使用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')
)

    输出：

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

相关问题