我觉得这应该很简单,但我很难用它。假设我有这个:
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分钟)。
熊猫的做法是什么?谢谢!
答案 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