在熊猫数据帧python中寻找价值

Question

我有一个包含三列的数据框

    A       B     C
    One     2     1
    Two     3     0.5
    Three   6     7

我有一个代表第二和第三列的列表： [（3,0.5），（6,7），（2,1）]

基于该列表，我想以列表的形式检索第一列中的值： [二，三，一]

我该怎么做？ 提前谢谢!!

Answer 1

您可以将列B和C设置为多索引，然后使用列表进行查询：

idx = [(3,0.5),(6,7),(2,1)]
df.set_index(['B', 'C']).A.loc[idx].values
# array(['Two', 'Three', 'One'], dtype=object)

如果您需要结果列表，请使用tolist作为@ Jezrael的评论。