我正在尝试对dask
和pandas
的性能进行基准测试。
def make_pandas(n):
df = pd.DataFrame(np.random.randint(10, size=(n, 3)))
return df
def make_dask(n):
df = da.from_array(np.random.randint(10, size=(n, 3)), chunks=10)
return df
def make_numpy(n):
return np.random.randint(10, size=(n, 3))
def sum_pandas(x): return x[0].sum()
def sum_dask(x): return x[1].sum()
def sum_numpy(x): return x[2].sum()
perfplot.show(
setup=lambda n: [make_pandas(n), make_dask(n), make_numpy(n)],
kernels=[sum_pandas, sum_dask, sum_numpy],
n_range=[2**k for k in range(2, 15)],
equality_check=False,
xlabel='len(df)')
有人可以解释结果吗?
将块增加到1000、8000和10000分别得到这些:
dask
是否应该并行化并随着大小的增加而变得更好?
答案 0 :(得分:0)
chunks
关键字是存储块大小的缩写,不是存储块的数量