python - 类似 index.get_loc 的函数，但用于多个值

Question

我有以下数据框:

df = pd.DataFrame({'color': ['Yellow', 'Green', 'Red', 'Orange'], 'weight': [0.5, 4, 1, 10]}, index = ['Banana','Melon','Apple','Pumpkin'])

看起来像这样:

Banana  Yellow     0.5
Melon    Green     4.0
Apple      Red     1.0
Pumpkin  Orange    10.0

我要做的是访问“香蕉”、“甜瓜”、“苹果”、“南瓜”子集的整数位置。我知道 pandas.index.get_loc 是对索引的单个元素执行此操作的一种方式。例如:df.index.get_loc('Apple') 返回 2。但是，我想一次对 multiple 元素执行此操作。除其他外，我尝试了 df.index.get_loc('Apple','Pumpkin')，希望得到列表 [2,3]，但这给出了以下错误:ValueError: Invalid fill method。期望填充 (ffill)、回填 (bfill) 或最近。得到南瓜

是否有一个函数可以让我一次获取多个整数位置？

Answer 1

尝试使用 get_indexer ，它接受类似输入的列表

df.index.get_indexer(['Apple','Pumpkin'])
Out[104]: array([2, 3], dtype=int32)