torcharrow.DataFrame.transform

DataFrame.transform(func: Callable, dtype: Optional[DType] = None, format: str = 'column', columns: Optional[List[str]] = None)

类似于 map()，但一次对行的迷你批次调用可调用对象。默认情况下，列作为 TorchArrow 列传递给可调用对象。如果 format=’python’，则输入将转换为 Python 类型。如果 format=’torch’，则输入将转换为 PyTorch 类型，如果结果类型 != 项目类型且可调用对象上缺少类型提示，则需要 dtype。