如何正确使用Pandas.sort_index()与searchsorted？降序排序问题求解

我需要对DataFrame/Series按索引排序，再用searchsorted查找目标值的iloc以完成切片操作。使用sort_index(ascending=True)升序排序时能得到预期结果，但设置为降序排序时，searchsorted无法返回正确结果。

升序排序正常示例 #

>>> s = pd.Series(['a', 'b', 'c', 'd'], index=[3, 2, 1, 4])
>>> s.sort_index(ascending=True, inplace=True)
>>> s
1    c
2    b
3    a
4    d
dtype: object
>>> s.index.searchsorted(1)
0    # 结果正常
>>> s.index.searchsorted(2)
1    # 结果正常
>>> s.index.searchsorted(3)
2    # 结果正常
>>> s.index.searchsorted(4)
3    # 结果正常

降序排序异常示例 #

>>> s.sort_index(ascending=False, inplace=True)
>>> s.index.searchsorted(1)
0    # 结果异常
>>> s.index.searchsorted(2)
0    # 结果异常
>>> s.index.searchsorted(3)
4    # 结果异常
>>> s.index.searchsorted(4)
4    # 结果异常

解决方法 #

核心原因是searchsorted默认假设索引是升序排列的，降序时必须调整参数匹配排序逻辑，以下是两种可行方案：

方案一：调用searchsorted时指定side参数 #

降序排序后，使用side='right'参数，让searchsorted适配降序的索引顺序：

s = pd.Series(['a', 'b', 'c', 'd'], index=[3, 2, 1, 4])
s.sort_index(ascending=False, inplace=True)

# 查找目标值1的正确位置
print(s.index.searchsorted(1, side='right'))  # 输出3，对应降序索引中1的iloc位置
# 查找目标值4的正确位置
print(s.index.searchsorted(4, side='right'))  # 输出0，对应降序索引中4的iloc位置

方案二：先升序获取位置再转换为降序位置 #

如果最终目的是切片，可以先在升序索引下完成searchsorted，再通过索引长度反转位置：

s = pd.Series(['a', 'b', 'c', 'd'], index=[3, 2, 1, 4])

# 升序排序获取目标位置
s_asc = s.sort_index(ascending=True)
pos_asc = s_asc.index.searchsorted(3)  # 输出2

# 转换为降序对应的位置
pos_desc = len(s_asc) - pos_asc  # 输出4-2=2

# 降序排序后执行切片
s_desc = s.sort_index(ascending=False)
print(s_desc.iloc[:pos_desc])
# 输出：
# 4    d
# 3    a
# dtype: object

补充说明：searchsorted的side参数控制插入位置的判定逻辑，当索引为降序时，side='right'会从右向左匹配符合条件的插入点，从而得到正确的iloc位置。

内容的提问来源于stack exchange，提问作者Leon