使用Dask数据框删除列

Question

这应该工作：

raw_data.drop('some_great_column', axis=1).compute(）

但是该列未删除。在大熊猫中，我使用：

raw_data.drop(['some_great_column'], axis=1, inplace=True)

但是就地在Dask中不存在。有任何想法吗？

Answer 1

您可以分为两个操作：

# dask operation
raw_data = raw_data.drop('some_great_column', axis=1)

# conversion to pandas
df = raw_data.compute()

然后将Pandas数据框导出到CSV文件：

df.to_csv(r'out.csv', index=False)