У меня есть массив dask (distMap) измерения (329960, 4). Я пытаюсь применить следующие операции, которые могут быть легко реализованы в NumPy, но я не могу реализовать в DASK.
d1_ix, d2_ix = np.argpartition(distMap, 2, axis=1)[:, :2].T
Любое предложение, как решить это?