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)