如何从索引和列已知的Pandas数据帧中获取多个值?

2024-05-07 23:29:57 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个像这样的熊猫数据框

>> 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。如何同时选择它们?在


Tags: 数据dataframepd
1条回答
网友
1楼 · 发布于 2024-05-07 23:29:57

使用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)

输出

^{pr2}$

相关问题 更多 >