torcharrow.functional.id_overlap_count

torcharrow.functional.id_overlap_count(input_ids: ListColumn, matching_ids: ListColumn)

Returns the number of overlaps between two lists of ids

Parameters:

• input_ids (First list of ids) –

• matching_ids (Second list of ids) –

Examples

>>> import torcharrow as ta
>>> from torcharrow import functional
>>> input_ids = ta.column([[1, 2, 3],[5,8],[13]])
>>> matching_ids = ta.column([[1],[2,3],[13]])
>>> functional.id_overlap_count(input_ids, matching_ids)
0  3
1  0
2  1
dtype: Float32(nullable=True), length: 3, null_count: 0