daskを使うときに `Bag.take（n）`で空の結果を避けるには？

Question

コンテキスト： Daskドキュメントでは、Bag.take()が最初のパーティションからのみ収集されることを明確に述べています。ただし、フィルタを使用すると、最初のパーティションは空で、それ以外のパーティションは空になることがあります。

質問： はそれはそれはnアイテム（またはnよりも小さい可能な最大）を収集するために、パーティションの十分な数から集めるようにBag.take()を使用することが可能です。

Answer 1

次のような何かができる：

は

from toolz import take 
f = lambda seq: list(take(n, seq)) 
b.reduction(f, f) 

これは、各パーティションの最初のn個の要素をつかむ一緒にそれらすべてを収集し、その結果の最初のn個の要素を取ります。