ray.data.Dataset.take

Dataset.take(limit: int = 20) → List[Dict[str, Any]]

从 Dataset 中返回最多 limit 行。

此方法有助于检查数据。

警告

take() 将最多 limit 行移动到调用者的机器上。如果 limit 很大，此方法可能会导致调用者出现 OutOfMemory 错误。

备注

此操作将触发对此数据集执行的延迟转换。

示例

>>> import ray
>>> ds = ray.data.range(100)
>>> ds.take(3)
[{'id': 0}, {'id': 1}, {'id': 2}]

时间复杂度: O(指定的限制)

参数:

limit – 返回的最大行数。

返回:

数据集中最多 limit 行的列表。

参见

take_all()

调用此方法以返回所有行。