在熊貓數據幀蟒

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的評論。