spx_yearly_returns_close_to_close_bokeh_vbar.py

import yfinance_download

from bokeh.plotting import figure, show
from bokeh.models   import NumeralTickFormatter, HoverTool
import bokeh.models

import bokeh.palettes
import bokeh.transform

df = yfinance_download.update_records('^GSPC', interval='1d')

# keep only the last record of each year

tmp = df.groupby(df.index.year).agg({'Close': 'last'})

tmp['Close_1'] = tmp['Close'].shift(1)

tmp['pct'] = (tmp['Close'] - tmp['Close_1']) / tmp['Close_1'] * 100

tmp.tail(40)

# create bokeh chart from tmp dataframe. x-axis is Date. y-axis is pct.

p = figure(
    title='S&P 500 Yearly Returns',
    x_axis_label='Year',
    y_axis_label='%',
    x_axis_type='datetime',
    sizing_mode='stretch_both'
)

p.vbar(
    x='Date',
    top='pct',
    source=tmp,
    line_color='black',
    line_width=1,
    alpha=0.5
)

p.xaxis.formatter = NumeralTickFormatter(format='0')
p.yaxis.formatter = NumeralTickFormatter(format='0')

p.add_tools(
    HoverTool(
        tooltips=[
            ('Year', '@Date'),
            ('%', '@pct{0.00}'),
        ],
        formatters={
            '@Date': 'datetime',
        },
        mode='vline'
    )
)

p.xaxis.ticker = bokeh.models.DatetimeTicker(desired_num_ticks=40)

show(p)