当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法:

Series.searchsorted(values, side='left', ascending=True, na_position='last')

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

参数

value框架(形状必须与自身一致)

假设要插入 Self 的值

sidestr {‘left’, ‘right’} 可选,默认为“左”

如果‘left’,则给出找到的第一个合适位置的索引如果‘right’,返回最后一个这样的索引

ascendingbool 可选,默认 True

Sorted Frame 是升序(否则降序)

na_positionstr {‘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.Series.searchsorted。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。