python - 在 Pandas 中，如何使用一个表中的值作为索引从另一个表中提取数据？

Question

我觉得这应该很简单，但我很难做到。假设我有这个:

df1:

ticker  hhmm <--- The hhmm value corresponds to the column in df2
======  ====
AAPL    0931
IBM     0930
XRX     1559

df2:

ticker  0930  0931  0932 ... 1559   <<---- 390 columns
======  ====  ====  ==== ... ====
AAPL    4.56  4.57  ...      ...     
IBM     7.98  ...   ...      ...
XRX     3.33  ...   ...      3.78

目标是在 df1 中创建一个值为 df2[df1['hhmm']] 的新列。

例如:

df1:

ticker  hhmm  df2val
======  ====  ======
AAPL    0931    4.57
IBM     0930    7.98
XRX     1559    3.78

两个 df 都将“ticker”作为索引，所以我可以简单地加入它们，但假设这使用了太多内存(我使用的数据帧比此处显示的示例大得多)。

我试过应用，它很慢(运行 15 分钟)。

执行此操作的 Pandas 方法是什么？谢谢!

Answer 1

有一个函数叫做lookup

df1['val']=df2.set_index('ticker').lookup(df1.ticker,df1.hhmm)
df1
Out[290]: 
  ticker  hhmm    val
0   AAPL  0931   4.57
1    IBM  0930   7.98
2    XRX  1559  33.00# I make up this number