torcharrow.DataFrame.filter

DataFrame.filter(predicate: Union[Callable, Iterable], columns: Optional[List[str]] = None)

选择谓词为 True 的行。与 Pandas 不同。使用 keep 进行 Pandas 过滤。

参数：

• 可迭代对象 (predicate - 可调用对象或) – 一个谓词函数或布尔值的迭代器，其长度与列相同。如果是 n 元谓词，请使用 columns 参数提供参数。

• 名称 (columns - 字符串列表) – 要使用哪些列来调用过滤器。如果为 None，则应用于所有列。

• None (默认值) – 要使用哪些列来调用过滤器。如果为 None，则应用于所有列。

另请参阅

map, reduce, flatmap

示例

>>> ta.column([1,2,3,4]).filter([True, False, True, False]) == ta.column([1,2,3,4]).filter(lambda x: x%2==1)
0  1
1  1
dtype: boolean, length: 2, null_count: 0