dask.bag.Bag.take

dask.bag.Bag.take

Bag.take(k, npartitions=1, compute=True, warn=True)[源代码]

取前 k 个元素。

参数
k整数

要返回的元素数量

npartitionsint, 可选

元素仅从第一个 npartitions 中提取,默认值为 1。如果第一个 npartitions 中的行数少于 k 行,则会发出警告并返回找到的所有行。传递 -1 以使用所有分区。

计算bool, 可选

是否计算结果,默认是 True。

警告bool, 可选

如果返回的元素数量少于请求的数量,是否发出警告,默认是 True。

>>> import dask.bag as db
>>> b = db.from_sequence(range(1_000))
>>> b.take(3)
(0, 1, 2)