python - 在 Pandas 数据框中选择行时如何保持顺序？

Question

我想按列表中给定的特定顺序选择行。例如

这个数据框

a=[['car',1],['bike',3],['jewel',2],['tv',5],['phone',6]]

df=pd.DataFrame(a,columns=['items','quantity'])

>>> df
   items  quantity
0    car         1
1   bike         3
2  jewel         2
3     tv         5
4  phone         6

我想获取顺序为 ['tv','car','phone'] 的行，即第一行 tv，然后是 car，最后是 phone。我试过这个方法，但它不保持秩序

arr=['tv','car','phone']

df.loc[df['items'].isin(arr)]

   items  quantity
0    car         1
3     tv         5
4  phone         6

Answer 1

这是一个使用 Index.get_indexer 的非侵入式解决方案这不涉及设置索引:

df.iloc[pd.Index(df['items']).get_indexer(['tv','car','phone'])]

   items  quantity
3     tv         5
0    car         1
4  phone         6

请注意，如果这将成为经常发生的事情(我指的是用列上的列表“索引”)，则最好将该列转换为索引。如果您对它进行排序，将获得加分。

df2 = df.set_index('items')
df2.loc[['tv','car','phone']]  

       quantity
items          
tv            5
car           1
phone         6