A Dask DataFrame is a large parallel DataFrame composed of many smaller pandas DataFrames, split along the index. These pandas DataFrames may live on disk for larger-than-memory computing on a single machine, or on many different machines in a cluster. One Dask DataFrame operation triggers many operations on the constituent pandas DataFrames.
Visit https://examples.dask.org/dataframe.html to see and run examples using Dask DataFrame.
Dask DataFrames coordinate many pandas DataFrames/Series arranged along the index. A Dask DataFrame is partitioned row-wise, grouping rows by index value for efficiency. These pandas objects may live on disk or on other machines.
Dask DataFrame copies the pandas DataFrame API¶
dask.DataFrame application programming interface (API) is a
subset of the
pd.DataFrame API, it should be familiar to pandas users.
There are some slight alterations due to the parallel nature of Dask:
>>> import dask.dataframe as dd >>> df = dd.read_csv('2014-*.csv') >>> df.head() x y 0 1 a 1 2 b 2 3 c 3 4 a 4 5 b 5 6 c >>> df2 = df[df.y == 'a'].x + 1 >>> df2.compute() 0 2 3 5 Name: x, dtype: int64
>>> import pandas as pd >>> df = pd.read_csv('2014-1.csv') >>> df.head() x y 0 1 a 1 2 b 2 3 c 3 4 a 4 5 b 5 6 c >>> df2 = df[df.y == 'a'].x + 1 >>> df2 0 2 3 5 Name: x, dtype: int64
As with all Dask collections, you trigger computation by calling the
Common Uses and Anti-Uses¶
Dask DataFrame is used in situations where pandas is commonly needed, usually when pandas fails due to data size or computation speed:
Manipulating large datasets, even when those datasets don’t fit in memory
Accelerating long computations by using many cores
Distributed computing on large datasets with standard pandas operations like groupby, join, and time series computations
Dask DataFrame may not be the best choice in the following situations:
If your dataset fits comfortably into RAM on your laptop, then you may be better off just using pandas. There may be simpler ways to improve performance than through parallelism
If your dataset doesn’t fit neatly into the pandas tabular model, then you might find more use in dask.bag or dask.array
If you need functions that are not implemented in Dask DataFrame, then you might want to look at dask.delayed which offers more flexibility
If you need a proper database with all that databases offer you might prefer something like Postgres
Dask DataFrame covers a well-used portion of the pandas API. The following class of computations works well:
- Trivially parallelizable operations (fast):
df.x + df.y,
df * df
df[df.x > 0]
df[df.x.isin([1, 2, 3])]
Date time/string accessors:
- Cleverly parallelizable operations (fast):
groupby-aggregate (with common aggregations):
groupby-apply on index:
df.groupby(['idx', 'x']).apply(myfunc), where
idxis the index level name
Join on index:
dd.merge(df1, df2, left_index=True, right_index=True)or
dd.merge(df1, df2, on=['idx', 'x'])where
idxis the index name for both
Join with pandas DataFrames:
dd.merge(df1, df2, on='id')
Element-wise operations with different partitions / divisions:
df1.x + df2.y
Date time resampling:
- Operations requiring a shuffle (slow-ish, unless on index, see Shuffling for GroupBy and Join)
groupby-apply not on index (with anything):
Join not on the index:
dd.merge(df1, df2, on='name')
However, Dask DataFrame does not implement the entire pandas interface. Users expecting this will be disappointed. Notably, Dask DataFrame has the following limitations:
Setting a new index from an unsorted column is expensive
Many operations like groupby-apply and join on unsorted columns require setting the index, which as mentioned above, is expensive
The pandas API is very large. Dask DataFrame does not attempt to implement many pandas features or any of the more exotic data structures like NDFrames
Operations that were slow on pandas, like iterating through row-by-row, remain slow on Dask DataFrame
See the DataFrame API documentation for a more extensive list.
By default, Dask DataFrame uses the multi-threaded scheduler. This exposes some parallelism when pandas or the underlying NumPy operations release the global interpreter lock (GIL). Generally, pandas is more GIL bound than NumPy, so multi-core speed-ups are not as pronounced for Dask DataFrame as they are for Dask Array. This is particularly true for string-heavy Python DataFrames, as Python strings are GIL bound.
There has been recent work on changing the underlying representation of pandas string data types to be backed by PyArrow Buffers, which should release the GIL, however, this work is still considered experimental.
When dealing with text data, you may see speedups by switching to the distributed scheduler either on a cluster or single machine.