torcharrow.DataFrame.filter

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

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

Parameters:

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

• names (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