在使用isin（）时，如何维护原始数据帧索引并保持输入列表的顺序？

Question

我有以下数据帧：g=pd。DataFrame（{'A'：[1,2,42,5,7]，'B'：[5,6,7,3,2]}）

    A  B
0   1  5
1   2  6
2  42  7
3   5  3
4   7  2

am使用以下列表筛选数据帧：

list_values = [5,7,1]

并使用以下命令获得以下输出：

indexes = g[g['A'].isin(list_values)].index.values

输出

array([0, 3, 4], dtype=int64)

如何更改代码，使索引如下所示？

array([3, 4, 0], dtype=int64)

本质上，我正在寻找一种方法来过滤具有列表的DF，并按照筛选列表的顺序返回原始索引值。

谢谢

我查看了这个，但并没有找到我要查找的内容：按列表的顺序从列表中选择数据帧的行

Answer 1

您可以使用orderedCategoricalDtype强制执行自定义排序顺序。排序后，返回5、7、1的所有索引。

import pandas as pd

my_cat = pd.CategoricalDtype(categories=list_values, ordered=True)
#CategoricalDtype(categories=[5, 7, 1], ordered=True)

g.loc[g['A'].isin(list_values), 'A'].astype(my_cat).sort_values().index
#Int64Index([3, 4, 0], dtype='int64')