在熊貓數據框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的注釋。