使用Pandas重新采样然后填充原始数据帧

时间:2016-06-30 15:37:07

标签: python pandas

我根据一周的收盘价和下周的开盘价来调查市场统计数据。要做到这一点,我在Pandas中使用resample。为了给出一个例子,我在下面使用pandas DataReader。

from pandas.io.data import DataReader

首先获取每日市场数据:

SP = DataReader("^GSPC", "yahoo") 
del SP['Adj Close'] 
del SP['Volume'] 

SP.head()

              Open       High         Low       Close
Date                
2010-01-04  1116.560059 1133.869995 1116.560059 1132.989990
2010-01-05  1132.660034 1136.630005 1129.660034 1136.520020

现在resample到每周时间范围:

ohlc_dict = {                                                                                                             
'Open':'first',                                                                                                                                                                                                         
'Close': 'last'}
w1_resamp = SP.resample('1w',how=ohlc_dict, closed='left', label='left')

这给了我每周关闭和打开的数据。我现在根据np.where声明突出显示上周关闭与本周开放之间的距离。

w1_resamp['distance'] = np.where(w1_resamp['Open'] < w1_resamp['Close'].shift(),(w1_resamp["Close"].shift() - w1_resamp["Open"]),'np.nan'); 



               Close    Open        distance
Date            
2010-01-03  1144.979980 1116.560059 
2010-01-10  1136.030029 1145.959961 
2010-01-17  1091.760010 1136.030029 
2010-01-24  1073.869995 1092.400024 
2010-01-31  1066.189941 1073.890015 
2010-02-07  1075.510010 1065.510010 0.6799310000001242
2010-02-14  1109.170044 1079.130005 
2010-02-21  1104.489990 1110.000000 
2010-02-28  1138.699951 1105.359985 
2010-03-07  1149.989990 1138.400024 0.29992700000002515
2010-03-14  1159.900024 1148.530029 1.4599610000000212

我现在想在原始数据框SP中添加一个新列,显示时间和日期差距(在w1_resamp['distance']中突出显示)已关闭但不知道如何执行此操作...任何人都可以提供帮助请?

根据评论中的请求,添加了在SP数据框中显示所需输出的图像:

desired output

1 个答案:

答案 0 :(得分:0)

我没有在“关闭差距”字段中关注您的请求,但是可以尝试一下,看看是否可以将其应用于索引以进行日期计算。

仅供参考,它看起来像是在描述“如何”方法,并打印警告以使用.apply()

import pandas as pd
import numpy as np

idx = pd.date_range("2018-01-01","2018-12-31")
columns = ['open','close']
data = np.random.normal(365,2)
df = pd.DataFrame(np.random.random((len(idx),len(columns))), columns = columns,index=idx)
df['high'] = df['open']*(1+np.random.uniform(.05, .20)) #bull market...
func = {
    'open': df['open'].resample('1w').first(),
    'close': df['close'].resample('1w').last(),
    'high': df['high'].resample('1w').max()
}

df_w = pd.DataFrame(func)
df_w['oc_diff'] = df_w['open'] - df_w['close'].shift()
df_w.head(10)

                open     close      high   oc_diff
2018-01-07  0.268054  0.352703  1.186531       NaN
2018-01-14  0.340011  0.907513  1.127548 -0.012693
2018-01-21  0.764949  0.907459  0.915084 -0.142564
2018-01-28  0.346734  0.703151  1.027472 -0.560725
2018-02-04  0.231348  0.960882  0.911420 -0.471803
相关问题