pandas.DataFrame.apply（）帶參數？

Question

我有一個投資組合數據幀與股票和數量的列表：

stock, quantity 
AAPL, 100 
GOOG, 50 

我有一個函數lookup(stock, date)返回的價格。

什麼是最優雅的方式來申請lookup()股票DataFrame？我想用DataFrame.apply()但我不相信我可以提供日期。

Answer 1

這應該工作：

def lookup(stock, date): 
    prices = {('AAPL', '2016-01-01'): 1, ('GOOG', '2016-01-01'): 2} 
    return prices[(stock, date)] 

>>> date = '2016-01-01' 
>>> df.apply(lambda x: lookup(x[0], date), axis=1) 
0 1 
1 2 
dtype: int64 

如果stock是索引，使用：

>>> df.apply(lambda x: lookup(x.name, date), axis=1)