Фильтровать строки по нескольким элементам, содержащимся в столбце списка в Polars?Python

Программы на Python
Ответить
Anonymous
 Фильтровать строки по нескольким элементам, содержащимся в столбце списка в Polars?

Сообщение Anonymous »

Для DataFrame со столбцом списка и списком элементов, которых нет во фрейме данных:
import polars as pl

df = pl.DataFrame({
"sets": [
[1, 2, 3],
[4],
[9, 10],
[2, 12],
[6, 6, 1],
[2, 0, 1],
[1, 1, 4],
[2, 7, 2],
]
})

items = [1, 2]

Есть ли эффективный способ отфильтровать таблицу, чтобы в ней были только строки, в которых значение столбца списка содержит а) один из элементов списка и б) все элементы списка?
Ожидаемый результат для ВСЕ:
shape: (2, 1)
┌───────────┐
│ sets │
│ --- │
│ list[i64] │
╞═══════════╡
│ [1, 2, 3] │
│ [2, 0, 1] │
└───────────┘

Ожидаемый результат для ЛЮБОГО:
shape: (6, 1)
┌───────────┐
│ sets │
│ --- │
│ list[i64] │
╞═══════════╡
│ [1, 2, 3] │
│ [2, 12] │
│ [6, 6, 1] │
│ [2, 0, 1] │
│ [1, 1, 4] │
│ [2, 7, 2] │
└───────────┘


Подробнее здесь: https://stackoverflow.com/questions/756 ... -in-polars
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»