torcharrow.Column.isin

Column.isin(values: Union[list, dict])

Check whether each element in the column is contained in values.

Parameters:

• array-like (values -) – Which values to check the presence of.

• dict (column or) – Which values to check the presence of.

Return type:

Column of booleans indicating if each element is in values.

Examples

>>> import torcharrow as ta
>>> df = ta.dataframe({'a': list(range(7)),
>>>             'b': list(reversed(range(7))),
>>>             'c': list(range(7))
>>>            })
>>> df[df['a'].isin([5])]
  index    a    b    c    d
-------  ---  ---  ---  ---
      0    5    1    5  104
dtype: Struct([Field('a', int64), Field('b', int64), Field('c', int64), Field('d', int64)]), count: 1, null_count: 0