我在python pandas上运行,不明白为什么窗口大小为40的滚动平均数没有与雅虎的股价图一起显示
首先,我得到了数据(包括通过的日期):
def get_data(dates):
"""Read stock data (adjusted close) for given symbols from CSV files."""
df = pd.DataFrame(index=dates)
df_temp = pd.read_csv('table.csv', index_col='Date', parse_dates=True,
usecols=['Date', 'Adj Close'], na_values=['nan'])
df = df.join(df_temp)
df.dropna()
return df
然后我找到滚动平均值(其中values=df(雅虎股票价格和窗口数据=40)):
^{pr2}$然后我去策划:
ax = df.plot(title="Bollinger Bands", label='YAHO')
rm_SPY.plot(label='Rolling mean', ax=ax)
最后,我只得到了一张雅虎收盘价的图表,没有滚动平均价,也没有其他人喜欢说的“移动点均线”
完整代码如下:
def get_data(dates):
"""Read stock data (adjusted close) for given symbols from CSV files."""
df = pd.DataFrame(index=dates)
df_temp = pd.read_csv('table.csv', index_col='Date', parse_dates=True,
usecols=['Date', 'Adj Close'], na_values=['nan'])
df = df.join(df_temp)
df.dropna()
return df
def get_rolling_mean(values, window):
"""Return rolling mean of given values, using specified window size."""
return values.rolling(center=False, window=window).mean()
def get_rolling_std(values, window):
"""Return rolling standard deviation of given values, using specified window
size."""
# TODO: Compute and return rolling standard deviation
return values.rolling(center=False, window=window).std()
def get_bollinger_bands(rm, rstd):
"""Return upper and lower Bollinger Bands."""
# TODO: Compute upper_band and lower_band
upper_band = rm+rstd
lower_band = rm-rstd
return upper_band, lower_band
def test_run():
# Read data
dates = pd.date_range('2012-01-01', '2012-12-31')
df = get_data(dates)
# Compute Bollinger Bands
# 1. Compute rolling mean
rm_SPY = get_rolling_mean(df, window=40)
# 2. Compute rolling standard deviation
rstd_SPY = get_rolling_std(df, window=40)
# 3. Compute upper and lower bands
upper_band, lower_band = get_bollinger_bands(rm_SPY, rstd_SPY)
# Plot raw SPY values, rolling mean and Bollinger Bands
ax = df.plot(title="Bollinger Bands", label='SPY')
rm_SPY.plot(label='Rolling mean', ax=ax)
upper_band.plot(label='upper band', ax=ax)
lower_band.plot(label='lower band', ax=ax)
# Add axis labels and legend
ax.set_xlabel("Date")
ax.set_ylabel("Price")
ax.legend(loc='upper left')
plt.show()
if __name__ == "__main__":
test_run()
请看这是否对您有帮助:
相关问题 更多 >
编程相关推荐