快捷方式

torcharrow.Column.filter

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

选择其中谓词为真的行。与 Pandas 不同。对于 Pandas 过滤,请使用 keep。

参数:
  • 可迭代对象 (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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源