Python 如何reshape 极地数据帧

发布于02月14日

我是新的极地，但已经与Pandas 和numpy相当多.假设我有一个类似这样的框架:

Column 1	Column 2	Column 3	Column 4	Column 5	Column 6
A	1	W	B	2	X
C	3	Y	D	4	Z

我想做一些类似df.rehape((-1，3))(In NumPy)的事情，以获得:

Column 1	Column 2	Column 3
A	1	W
B	2	X
C	3	Y
D	4	Z

然而，我还没能在极点中找到类似的函数.当然，在这种情况下df.melt()不起作用，执行df=pl.DataFrame(df.to_numpy().reshape((-1，3)似乎不是最佳解决方案.会很感激大家的帮助的！

翻阅了文件，但无济于事.

推荐答案

我不认为北极星在上面概述的情况下有专门的reshape功能.但是，您可以组合pl.concat来连接使用pl.DataFrame.select创建的数据帧片段.

pl.concat([
    df.select("Column 1", "Column 2", "Column 3"),
    df.select(
        pl.col("Column 4").alias("Column 1"),
        pl.col("Column 5").alias("Column 2"),
        pl.col("Column 6").alias("Column 3"),
    ),
])

Output.个

shape: (4, 3)
┌──────────┬──────────┬──────────┐
│ Column 1 ┆ Column 2 ┆ Column 3 │
│ ---      ┆ ---      ┆ ---      │
│ str      ┆ i64      ┆ str      │
╞══════════╪══════════╪══════════╡
│ A        ┆ 1        ┆ W        │
│ C        ┆ 3        ┆ Y        │
│ B        ┆ 2        ┆ X        │
│ D        ┆ 4        ┆ Z        │
└──────────┴──────────┴──────────┘