在Pandas中,如何使用一个表中的值作为索引从另一个表中提取数据?

时间:2018-03-22 20:59:28

标签: python pandas numpy dataframe

我觉得这应该很简单,但我很难用它。假设我有这个:

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 [&#39; hhmm&#39;]]。

例如:

df1:

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

两个df都有&#39; ticker&#39;作为他们的索引,所以我可以简单地加入它们但是假设它使用了太多的内存(我使用的数据帧比这里显示的示例大得多)。

我已经尝试过申请和它的懒散(运行15分钟)。

熊猫的做法是什么?谢谢!

3 个答案:

答案 0 :(得分:1)

这是您尝试做的最小例子。希望这能给你足够的提示:

# sample data
df1 = pd.DataFrame({'ticker':['AAPL','IBM','XRX'], 'hhmm':['0931','0930','1559']})

df2 = pd.DataFrame({'ticker':['AAPL','IBM','XRX'],
                    '0931': [2,2,3],
                    '0930': [5,6,7],
                    '1559': [8,7,6]})

# melt to match the format to join
df2 = pd.melt(df2, id_vars='ticker',var_name='hhmm',value_name='df2val')

# join to df1
df1.merge(df2, on=['ticker','hhmm'])

    hhmm    ticker  df2val
0   0931    AAPL    2
1   0930    IBM     6
2   1559    XRX     6

答案 1 :(得分: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

答案 2 :(得分:1)

尝试

df2.set_index('ticker').stack().loc[df1.apply(tuple, axis = 1)]

ticker      
AAPL    931     4.57
IBM     930     7.98
XRX     1559    3.78