python - 使用 Dask DataFrame loc 更改列值

Question

我有一个庞大的数据库，我需要根据特定条件更改列的值。

在 Pandas 中，我执行以下代码来完成我想要的：

df.loc[
        (df['ID_CRITERIO_APURACAO'] == TipoDestinatario.RESIDENCIAL.value) &
        (df['CODG_GRUPO_TENSAO'] == 8) &
        (df['CONSUMO'].between(0, 30)),
        'DESCONTO'
    ] = 35

我怎样才能在 Dask 中做类似的事情？

Answer 1

Dask不支持就地突变。尝试这个：

condition = (df['ID_CRITERIO_APURACAO'] == TipoDestinatario.RESIDENCIAL.value) &
    (df['CODG_GRUPO_TENSAO'] == 8) &
    (df['CONSUMO'].between(0, 30))

desconto = df.where(condition, 35)
df['DESCONTO'] = desconto