如何将空的pandas数据帧转换为polars数据帧

Question

我定义了一个pandas DataFrame，如下所示：

df_tmp = pd.DataFrame({'EDT': pd.Series(dtype='datetime64[ns]'),
'FSPB': pd.Series(dtype='str'),
'FS_LA': pd.Series(dtype='str'),
'lA': pd.Series(dtype='int'),
'avg': pd.Series(dtype='float64'),
'nw': pd.Series(dtype='float64')})

有没有办法把上面的数据转换成一个空的polars数据帧？

Answer 1

根据polars文档，polars DataFrame可以在其构造函数中使用pandas DataFrame，因此：

import pandas as pd
import polars as pl
df_tmp = pd.DataFrame({'EDT': pd.Series(dtype='datetime64[ns]'),
'FSPB': pd.Series(dtype='str'),
'FS_LA': pd.Series(dtype='str'),
'lA': pd.Series(dtype='int'),
'avg': pd.Series(dtype='float64'),
'nw': pd.Series(dtype='float64')})
df = pl.DataFrame(df_tmp)

应该起作用。

Answer 2

import polars as pl
import pandas as pd
pandas_df = pd.DataFrame({'EDT': pd.Series(dtype='datetime64[ns]'),
'FSPB': pd.Series(dtype='str'),
'FS_LA': pd.Series(dtype='str'),
'lA': pd.Series(dtype='int'),
'avg': pd.Series(dtype='float64'),
'nw': pd.Series(dtype='float64')})
pl.from_pandas(pandas_df)

shape: (0, 6)
┌──────────────┬──────┬───────┬─────┬─────┬─────┐
│ EDT          ┆ FSPB ┆ FS_LA ┆ lA  ┆ avg ┆ nw  │
│ ---          ┆ ---  ┆ ---   ┆ --- ┆ --- ┆ --- │
│ datetime[ns] ┆ str  ┆ str   ┆ i64 ┆ f64 ┆ f64 │
╞══════════════╪══════╪═══════╪═════╪═════╪═════╡
└──────────────┴──────┴───────┴─────┴─────┴─────┘