主页 / coding / 问题 / 79006642
Accepted
Andi
Asked: 2024-09-20 19:58:52 +0800 CST

将极坐标数据框中列表列的元素与常规 python 列表的元素相乘

  • 772

我有一个pl.DataFrame包含如下列表的列：

import polars as pl

df = pl.DataFrame(
    {
        "symbol": ["A", "A", "B", "B"],
        "roc": [[0.1, 0.2], [0.3, 0.4], [0.5, 0.6], [0.7, 0.8]],
    }
)

shape: (4, 2)
┌────────┬────────────┐
│ symbol ┆ roc        │
│ ---    ┆ ---        │
│ str    ┆ list[f64]  │
╞════════╪════════════╡
│ A      ┆ [0.1, 0.2] │
│ A      ┆ [0.3, 0.4] │
│ B      ┆ [0.5, 0.6] │
│ B      ┆ [0.7, 0.8] │
└────────┴────────────┘

此外，我有一个常规的 python 列表weights = [0.3, 0.7]

pl.col("roc")将乘以的一种有效方法是什么？weights即将 列的第一个和第二个元素分别与 的第一个和第二个元素相乘weights？

预期输出如下：

shape: (4, 3)
┌────────┬────────────┐──────────────┐
│ symbol ┆ roc        │ roc_wgt      │
│ ---    ┆ ---        │ ---          │
│ str    ┆ list[f64]  │ list[f64]    │
╞════════╪════════════╡══════════════╡
│ A      ┆ [0.1, 0.2] │ [0.03, 0.14] │ = [0.1 * 0.3, 0.2 * 0.7]
│ A      ┆ [0.3, 0.4] │ [0.09, 0.28] │ = [0.3 * 0.3, 0.4 * 0.7]
│ B      ┆ [0.5, 0.6] │ [0.15, 0.42] │ = [0.5 * 0.3, 0.6 * 0.7]
│ B      ┆ [0.7, 0.8] │ [0.21, 0.56] │ = [0.7 * 0.3, 0.8 * 0.7]
└────────┴────────────┘──────────────┘
python

4 个回答

  1. Best Answer

    有一个待处理的 PR 允许pl.col.roc * pl.lit(weights)

    还有Array固定宽度类型。

    dtype = pl.Array(float, 2)

(df.with_columns(wgt = pl.lit(weights, dtype))
   .with_columns(roc_wgt = pl.col.roc.cast(dtype) * pl.col.wgt)
)

    shape: (4, 4)
┌────────┬────────────┬───────────────┬───────────────┐
│ symbol ┆ roc        ┆ wgt           ┆ roc_wgt       │
│ ---    ┆ ---        ┆ ---           ┆ ---           │
│ str    ┆ list[f64]  ┆ array[f64, 2] ┆ array[f64, 2] │
╞════════╪════════════╪═══════════════╪═══════════════╡
│ A      ┆ [0.1, 0.2] ┆ [0.3, 0.7]    ┆ [0.03, 0.14]  │
│ A      ┆ [0.3, 0.4] ┆ [0.3, 0.7]    ┆ [0.09, 0.28]  │
│ B      ┆ [0.5, 0.6] ┆ [0.3, 0.7]    ┆ [0.15, 0.42]  │
│ B      ┆ [0.7, 0.8] ┆ [0.3, 0.7]    ┆ [0.21, 0.56]  │
└────────┴────────────┴───────────────┴───────────────┘

    由于目前情况紧急，因此需要添加一列pl.lit()。

    • 4

  2. 首先运行的一个选项explode：

    (df.with_row_index().with_columns(l=weights).explode(['roc', 'l'])
   .with_columns(roc_wgt=pl.col('roc')*pl.col('l'))
   .group_by('index')
   .agg(pl.col('symbol').first(),
        pl.col('roc'),
        pl.col('roc_wgt'))
)

    输出：

    ┌───────┬────────┬────────────┬──────────────┐
│ index ┆ symbol ┆ roc        ┆ roc_wgt      │
│ ---   ┆ ---    ┆ ---        ┆ ---          │
│ u32   ┆ str    ┆ list[f64]  ┆ list[f64]    │
╞═══════╪════════╪════════════╪══════════════╡
│ 0     ┆ A      ┆ [0.1, 0.2] ┆ [0.03, 0.14] │
│ 1     ┆ A      ┆ [0.3, 0.4] ┆ [0.09, 0.28] │
│ 2     ┆ B      ┆ [0.5, 0.6] ┆ [0.15, 0.42] │
│ 3     ┆ B      ┆ [0.7, 0.8] ┆ [0.21, 0.56] │
└───────┴────────┴────────────┴──────────────┘

    或者，使用map_elements：

    import numpy as np

df.with_columns(roc_wgt=pl.col('roc')
                          .map_elements(lambda x: x*np.array(weights),
                                        return_dtype=pl.List(pl.Float64)))

    输出：

    ┌────────┬────────────┬──────────────┐
│ symbol ┆ roc        ┆ roc_wgt      │
│ ---    ┆ ---        ┆ ---          │
│ str    ┆ list[f64]  ┆ list[f64]    │
╞════════╪════════════╪══════════════╡
│ A      ┆ [0.1, 0.2] ┆ [0.03, 0.14] │
│ A      ┆ [0.3, 0.4] ┆ [0.09, 0.28] │
│ B      ┆ [0.5, 0.6] ┆ [0.15, 0.42] │
│ B      ┆ [0.7, 0.8] ┆ [0.21, 0.56] │
└────────┴────────────┴──────────────┘
    • 1

  3. 您还可以遍历权重列表，从列中获取相应的元素，将它们相乘并将这些结果连接在一起形成一个新列表。

    import polars as pl

df = pl.DataFrame(
    {
        "symbol": ["A", "A", "B", "B"],
        "roc": [[0.1, 0.2], [0.3, 0.4], [0.5, 0.6], [0.7, 0.8]],
    }
)

weights = [.3, .7]

print(
    df.with_columns(
        roc_wgt=pl.concat_list(
            pl.col('roc').list.get(i) * wgt for i, wgt in enumerate(weights)
        )
    )
)
# shape: (4, 3)
# ┌────────┬────────────┬──────────────┐
# │ symbol ┆ roc        ┆ roc_wgt      │
# │ ---    ┆ ---        ┆ ---          │
# │ str    ┆ list[f64]  ┆ list[f64]    │
# ╞════════╪════════════╪══════════════╡
# │ A      ┆ [0.1, 0.2] ┆ [0.03, 0.14] │
# │ A      ┆ [0.3, 0.4] ┆ [0.09, 0.28] │
# │ B      ┆ [0.5, 0.6] ┆ [0.15, 0.42] │
# │ B      ┆ [0.7, 0.8] ┆ [0.21, 0.56] │
# └────────┴────────────┴──────────────┘
    • 1

  4. 不幸的是，polars 目前还不支持对列表进行此类操作，但它支持对结构进行此类操作：

    (
    df.with_columns(
        roc_wgt = pl.col.roc.list.to_struct() * pl.lit(weights).list.to_struct()
    ).with_columns(
        roc_wgt = pl.concat_list(pl.col.roc_wgt.struct.field("*"))
    )
)

    shape: (4, 3)
┌────────┬────────────┬──────────────┐
│ symbol ┆ roc        ┆ roc_wgt      │
│ ---    ┆ ---        ┆ ---          │
│ str    ┆ list[f64]  ┆ list[f64]    │
╞════════╪════════════╪══════════════╡
│ A      ┆ [0.1, 0.2] ┆ [0.03, 0.14] │
│ A      ┆ [0.3, 0.4] ┆ [0.09, 0.28] │
│ B      ┆ [0.5, 0.6] ┆ [0.15, 0.42] │
│ B      ┆ [0.7, 0.8] ┆ [0.21, 0.56] │
└────────┴────────────┴──────────────┘
    • 0

相关问题