當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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


用法:

DataFrame.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.DataFrame.searchsorted。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。