主页 / coding / 问题 / 77354114
Accepted
Thomas
Asked: 2023-10-25 01:14:15 +0800 CST

如何使用本机表达式 API 在 Polars 中生成项目组合？

  • 772

有没有一种方法可以在 Polars 列内的列表中生成项目组合，而无需为每行使用.map_elements()+ itertools？

这是我当前的解决方案：

import polars as pl
import itertools

(pl.DataFrame({'col': [['a', 'b', 'c']]})
   .with_columns(pl.col('col')
                   .map_elements(lambda list_o_things: [sorted((thing_1, thing_2))
                                                        for thing_1, thing_2 
                                                        in itertools.combinations(list_o_things, 2)])
                )
)

返回这个：

[['a', 'b'], ['a', 'c'], ['b', 'c']]

list

1 个回答

  1. Best Answer

    分解嵌套结构，与其自身进行交叉连接，过滤掉冗余条目，连接到列表，并内爆到嵌套列表。

    df=pl.DataFrame({'col': [['a', 'b', 'c']]})
(
    df
    .explode('col')
    .join(
        df.explode('col'), how='cross')
    .filter(pl.col('col')<pl.col('col_right'))
    .select(pl.concat_list('col','col_right').implode())
    )
shape: (1, 1)
┌──────────────────────────────────────┐
│ col                                  │
│ ---                                  │
│ list[list[str]]                      │
╞══════════════════════════════════════╡
│ [["a", "b"], ["a", "c"], ["b", "c"]] │
└──────────────────────────────────────┘
    • 1

相关问题