如何使用cudf删除带有NA的列？

Question

熊猫：

data = data.dropna(axis = 'columns')

我正在尝试使用cudf dataframe执行类似的操作，但apis不提供此功能。

我的解决方案是转换为pandas df，执行上述命令，然后重新转换为cudf。有没有更好的解决方案？

Answer 1

cuDF现在支持基于列的dropna，因此可以执行以下操作：

import cudf
​
df = cudf.DataFrame({'a':[0,1,None], 'b':[None,0,2], 'c':[1,2,3]})
print(df)
      a     b  c
0     0  null  1
1     1     0  2
2  null     2  3

df.dropna(axis='columns')
    c
0   1
1   2
2   3

Answer 2

在实现dropna之前，您可以检查每一列的null_count并使用null_count>0删除它们。