python - 在 Pandas 数据框中获取索引作为 int

Question

我有一个由字符串索引的 Pandas 数据框。假设我的索引看起来像 df.index = ['AA','AB','AC',...] 并且我想访问 df.loc['AC': 'AE']，效果很好。

有什么方法可以得到这些索引的位置，给我 ['AC':'AE'] => [2,3,4]？我知道有 df.index.get_loc('AC') => 2 但这只适用于单个值而不适用于列表。

Answer 1

使用:

df = pd.DataFrame({'a': [5,6,7,8, 10]}, index=['AA','AB','AC','AD','AE'])

pos = list(range(df.index.get_loc('AC'), df.index.get_loc('AE') + 1))
print (pos)
[2, 3, 4]

另一个解决方案 Index.searchsorted :

pos = list(range(df.index.searchsorted('AC'), df.index.searchsorted('AE') + 1))
print (pos)
[2, 3, 4]

---

a = df.index.searchsorted(['AC', 'AE'])
pos = list(range(a[0], a[1] + 1))
print (pos)
[2, 3, 4]