Wie vermeidet man ein leeres Ergebnis mit `Bag.take (n)` bei Verwendung von DASK?

Question

Kontext: Dask Dokumentation besagt eindeutig, dass Bag.take() nur von der ersten Partition sammeln. Bei Verwendung eines Filters kann es jedoch vorkommen, dass die erste Partition leer ist, andere nicht.

Frage: Ist es möglich, Bag.take() zu verwenden, so dass es von einer ausreichenden Anzahl von Partitionen sammelt (verfügbar weniger oder maximal als als n) die n Gegenstände zu sammeln.

Answer 1

Man könnte so etwas wie folgt tun:

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

Dieser greift die ersten n Elemente jeder Partition, sammelt sie alle zusammen, und nimmt dann die ersten n Elemente des Ergebnisses.