得到一個(gè)像下面這樣的數(shù)據(jù)框，它的索引是 [date,assetName]如何在 10 天內(nèi)添加帶有開盤價(jià)的新列？日期索引列的數(shù)據(jù)類型是時(shí)間戳（'2007-02-01 00:00:00+0000', tz='UTC'）from datetime import datetimeimport pandas as pddatetime_object = datetime.strptime('Feb 1 2007', '%b %d %Y')market_df = pd.DataFrame({'date': [ datetime.strptime('Feb 1 2007', '%b %d %Y') ,datetime.strptime('Feb 1 2007', '%b %d %Y') ,datetime.strptime('Feb 1 2007', '%b %d %Y') ,datetime.strptime('Feb 11 2007', '%b %d %Y') ,datetime.strptime('Feb 11 2007', '%b %d %Y') ,datetime.strptime('Feb 11 2007', '%b %d %Y') ], 'assetName': ['AAPL', 'GOOGL', 'AMZN','AAPL', 'GOOGL', 'AMZN'], 'open': [20, 30.9, 40, 22.3, 35, 45], 'close': [20.5, 30.6, 40.5,22.3,35,45], })market_df = market_df.set_index(['date','assetName']) 如何使用新列“price_in_10_days”獲取此數(shù)據(jù)框date assetName open close price_in_10_days2007-02-01 AAPL 20.0 20.5 22.32007-02-01 GOOGL 30.9 30.6 35.02007-02-01 AMZN 40.0 40.5 45.02007-02-11 AAPL 22.3 22.3 NaN2007-02-11 GOOGL 35.0 35.0 NaN2007-02-11 AMZN 45.0 45.0 NaN
1 回答
qq_花開花謝_0
TA貢獻(xiàn)1835條經(jīng)驗(yàn) 獲得超7個(gè)贊
我相信你需要?jiǎng)?chuàng)建DatetimeIndexby unstack，然后shift是10幾天和最后stack：
market_df['price_in_10_days'] = market_df['open'].unstack().shift(-1, freq='10D').stack()
print (market_df)
open close price_in_10_days
date assetName
2007-02-01 AAPL 20.0 20.5 22.3
GOOGL 30.9 30.6 35.0
AMZN 40.0 40.5 45.0
2007-02-11 AAPL 22.3 22.3 NaN
GOOGL 35.0 35.0 NaN
AMZN 45.0 45.0 NaN
添加回答
舉報(bào)
0/150
提交
取消