python - 在 Pandas 数据框 python 中寻找值(value)

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 的评论。