使用 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

我正在使用以下列表来过滤数据框：

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，并按照过滤器列表的顺序返回原始索引值。

谢谢！

我查看了这个，但没有找到我要找的东西： Select rows of pandas dataframe from list, in order of list

Answer 1

您可以使用有序的CategoricalDtype 来强制执行自定义排序顺序。排序后返回 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')