Установка фрагмента столбца в список значений в кадре данных Polars ⇐ Python

[Anonymous]

В приведенном ниже коде я создаю фрейм данных Polars и Pandas с идентичными данными. Я хочу выбрать набор строк на основе условия в столбце A, а затем обновить соответствующие строки для столбца C. Я описал, как я бы сделал это с фреймом данных pandas, но не понимаю, как заставить это работать с полярами. Самое близкое, что я получил, — это использование «когда-то-иначе», но я не могу использовать ничего, кроме одного значения в классе then.
import pandas as pd
import polars as pl

df_pd = pd.DataFrame({'A': ['x', 'x', 'x', 'x', 'y', 'y', 'y', 'y'],
'B': [1, 1, 2, 2, 1, 1, 2, 2],
'C': [1, 2, 3, 4, 5, 6, 7, 8]})

df_pl = pl.DataFrame({'A': ['x', 'x', 'x', 'x', 'y', 'y', 'y', 'y'],
'B': [1, 1, 2, 2, 1, 1, 2, 2],
'C': [1, 2, 3, 4, 5, 6, 7, 8]})

df_pd.loc[df_pd['A'] == 'x', 'C'] = [-1, -2, -3, -4]

df_pl ???



Подробнее здесь: https://stackoverflow.com/questions/792 ... -dataframe