python-3.x - 使用索引键值调用 DataFrame 索引列表

Question

df = pd.DataFrame([[3,3,3]]*4,index=['a','b','c','d'])

虽然我们可以通过指定行号来提取索引部分的副本，如下所示: i1=df.index[1:3].copy()

不幸的是，我们无法通过指定键来提取索引部分的副本(就像 df.loc 方法的情况)。当我尝试以下操作时:

i2=df.index['a':'c'].copy()

我收到以下错误:

TypeError: slice indices must be integers or None or have an __index__ method

有没有其他方法可以根据键调用索引的子集？谢谢

Answer 1

最简单的是loc使用索引:

i1 = df.loc['b':'c'].index
print (i1)
Index(['b', 'c'], dtype='object')

或者可以使用 get_loc职位:

i1 = df.index

i1 = i1[i1.get_loc('b') : i1.get_loc('d') + 1]
print (i1)
Index(['b', 'c'], dtype='object')

---

i1 = i1[i1.get_loc('b') : i1.get_loc('d') + 1]
print (i1)
Index(['b', 'c', 'd'], dtype='object')

备选方案:

i1 = i1[i1.searchsorted('b') : i1.searchsorted('d') + 1]
print (i1)
Index(['b', 'c', 'd'], dtype='object')