Shortcuts

torcharrow.functional.has_id_overlap

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

如果两个输入列重叠,则返回1.0,否则返回0.0

Parameters:
  • input_ids (第一个ID列表) –

  • matching_ids (第二个ID列表) –

示例

>>> 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.has_id_overlap(input_ids, matching_ids)
0  1
1  0
2  1
dtype: Float32(nullable=True), length: 3, null_count: 0