dask - 无法使用 dask 删除列或切片数据框？

Question

我正在尝试使用 dask 而不是 Pandas，因为我有 2.6gb csv 文件。
我加载它，我想删除一列。但似乎无论是 drop 方法
df.drop('column') 或切片 df[ : , :-1]

尚未实现。是这种情况还是我只是遗漏了什么？

Answer 1

我们实现了 drop this PR 中的方法.这从 dask 0.7.0 开始可用。

In [1]: import pandas as pd

In [2]: df = pd.DataFrame({'x': [1, 2, 3], 'y': [3, 2, 1]})

In [3]: import dask.dataframe as dd

In [4]: ddf = dd.from_pandas(df, npartitions=2)

In [5]: ddf.drop('y', axis=1).compute()
Out[5]: 
   x
0  1
1  2
2  3

以前也可以使用列名切片；当然，如果您有很多列，这可能不那么有吸引力。

In [6]: ddf[['x']].compute()
Out[6]: 
   x
0  1
1  2
2  3