Преобразование столбца кадра данных cuDF в 1 или 0 для значений «истина» / «ложь»

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

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


person rnyai    schedule 22.08.2019    source источник


Ответы (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
person Nick Becker    schedule 22.08.2019