Python cudf.DataFrame.searchsorted用法及代码示例

用法: DataFrame.searchsorted(values, side='left', ascending=True, na_position='last')

查找应插入元素以保持顺序的索引

参数：

value：框架(形状必须与自身一致): 假设要插入 Self 的值
side：str {‘left’, ‘right’} 可选，默认为“左”: 如果‘left’，则给出找到的第一个合适位置的索引如果‘right’，返回最后一个这样的索引
ascending：bool 可选，默认 True: Sorted Frame 是升序(否则降序)
na_position：str {‘last’, ‘first’} 可选，默认‘last’: 空值在排序顺序中的位置

插入点的一维 Cupy 数组

例子：

>>> s = cudf.Series([1, 2, 3])
>>> s.searchsorted(4)
3
>>> s.searchsorted([0, 4])
array([0, 3], dtype=int32)
>>> s.searchsorted([1, 3], side='left')
array([0, 2], dtype=int32)
>>> s.searchsorted([1, 3], side='right')
array([1, 3], dtype=int32)

如果值不是单调排序的，则可能会返回错误的位置：

>>> s = cudf.Series([2, 1, 3])
>>> s.searchsorted(1)
0   # wrong result, correct would be 1

>>> df = cudf.DataFrame({'a': [1, 3, 5, 7], 'b': [10, 12, 14, 16]})
>>> df
   a   b
0  1  10
1  3  12
2  5  14
3  7  16
>>> values_df = cudf.DataFrame({'a': [0, 2, 5, 6],
... 'b': [10, 11, 13, 15]})
>>> values_df
   a   b
0  0  10
1  2  17
2  5  13
3  6  15
>>> df.searchsorted(values_df, ascending=False)
array([4, 4, 4, 0], dtype=int32)

相关用法

注：本文由纯净天空筛选整理自rapids.ai大神的英文原创作品 cudf.DataFrame.searchsorted。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

例子：