Объедините два столбца в кортеж в API Polars Python ⇐ Python

[Anonymous]

У меня есть кадр данных Polars, который выглядит так:

Код: Выделить всё

import polars as pl

df = pl.DataFrame({"bid": [1, 2, 3], "fid": [4, 5, 6]})

Я хотел бы объединить два столбца по строкам в кортеж, чтобы результат выглядел так:

Код: Выделить всё

pl.DataFrame({"bfid": [(1, 4), (2, 5), (3, 6)]})

Код: Выделить всё

shape: (3, 1)
┌───────────┐
│ bfid      │
│ ---       │
│ list[i64] │
╞═══════════╡
│ [1, 4]    │
│ [2, 5]    │
│ [3, 6]    │
└───────────┘

Я пытался сделать это: df2.with_columns(pl.map(['bid', 'fid'], лямбда x: (x[0], x[1]))) что неправильно, но также довольно медленно, если я пытаюсь масштабироваться до больших наборов данных.


Подробнее здесь: https://stackoverflow.com/questions/741 ... python-api