Преобразовать столбец фрейма cuDF в 1 или 0 для значений "истина" / "ложь"

Question

Я использую док-контейнер RAPIDS (выпуск 0.9). Как я могу сделать следующее с RAPIDS cuDF?

df['new_column'] = df['column_name'] > condition df[['new_column']] *= 1

Answer 1

Вы можете сделать это так же, как с пандами.

import cudf

df = cudf.DataFrame({'a':[0,1,2,3,4]})
df['new'] = df['a'] >= 3
df['new'] = df['new'].astype('int') # could use int8, int32, or int64

# could also do (df['a'] >= 3).astype('int')
df

    a   new
0   0   0
1   1   0
2   2   0
3   3   1
4   4   1