如何从已知索引和列的 Pandas 数据框中获取多个值？ [复制]

Question

我有一个看起来像这样的 Pandas 数据框

>> pd.DataFrame({'A': [1,2,3], 'B':[4,5,6], 'C':[7,8,9]})

     A   B   C  
0    1   4   7  
1    2   5   8  
2    3   6   9  

我想选择与索引列对 (0,'A'),(2,'B'),(1,'C') 对应的值 1,6,8。如何同时选择它们？

Answer 1

使用lookup:

import pandas as pd

df = pd.DataFrame({'A': [1,2,3], 'B':[4,5,6], 'C':[7,8,9]})

rows, cols = zip(*[(0,'A'),(2,'B'),(1,'C')])

result = df.lookup(rows, cols)
print(result)

输出

[1 6 8]