从列表中选择熊猫数据表行，按列表的顺序

Question

df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]})
df

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

# from
list_of_values = [3,4,6]

# we would like, as a result
#      A   B
# 2    3   3
# 3    4   5
# 1    6   2

Answer 1

克服这种情况的一种方法是使'a'列成为索引并在新生成的pandas.dataframe上使用loc。最后，可以重置子采样数据帧的索引。

以下是如何：

ret = df.set_index('A').loc[list_of_values].reset_index(inplace=False)

# ret is
#      A   B
# 0    3   3
# 1    4   5
# 2    6   2 

请注意，这种方法的缺点是在过程中丢失了原始索引。