| from plotly.subplots import make_subplots |
| import plotly.graph_objects as go |
|
|
|
|
| def plot_multiple_columns(dataframe, primary_column, secondary_column, ternary_column=None, scale_similarity=None): |
|
|
| figure = make_subplots(specs=[[{"secondary_y": True}]]) |
|
|
| figure.add_trace(go.Scatter(x=dataframe.index, y=dataframe[primary_column], name=primary_column, |
| line=dict(color='#FF0000', width=1)), secondary_y=False) |
| figure.add_trace(go.Scatter(x=dataframe.index, y=dataframe[secondary_column], name=secondary_column, |
| line=dict(color='#00FF00', width=1)), secondary_y=True) |
|
|
| if ternary_column is not None: |
| if scale_similarity == 1: |
| figure.add_trace(go.Scatter(x=dataframe.index, y=dataframe[ternary_column], name=ternary_column, |
| line=dict(color='#0000FF', width=1)), secondary_y=False) |
| elif scale_similarity == 2: |
| figure.add_trace(go.Scatter(x=dataframe.index, y=dataframe[ternary_column], name=ternary_column, |
| line=dict(color='#0000FF', width=1)), secondary_y=True) |
|
|
| figure.update_layout(title_text=f"{primary_column} vs {secondary_column}") |
| figure.update_xaxes(title_text="time") |
|
|
| return figure |
|
|
|
|
| def plot_volume_profile(main_dataframe, volume_profile): |
|
|
| figure = go.Figure(data=[go.Bar(x=volume_profile.Volume, y=volume_profile.Close, |
| orientation="h", xaxis="x", yaxis="y", visible=True, name='Volume Profile', |
| showlegend=False), |
| go.Candlestick(x=main_dataframe.index, open=main_dataframe.open, high=main_dataframe.high, |
| low=main_dataframe.low, close=main_dataframe.close, xaxis="x2", yaxis="y", |
| visible=True, showlegend=False, name='OHLCV')], |
| layout=go.Layout(title=go.layout.Title(text="Candlestick With Volume Profile"), |
| autosize=False, width=700, height=550, |
| xaxis=go.layout.XAxis(side="top", |
| showticklabels=False, |
| rangeslider=go.layout.xaxis.Rangeslider(visible=False)), |
| yaxis=go.layout.YAxis(side="left", showticklabels=False), |
| xaxis2=go.layout.XAxis(side="bottom", title="Date", |
| overlaying="x", |
| rangeslider=go.layout.xaxis.Rangeslider(visible=False)), |
| |
| |
| )) |
|
|
| figure.data[0].marker.color = '#999900' |
| figure.data[1].increasing.fillcolor = 'rgb(230,72,72)' |
| figure.data[1].increasing.line.color = 'rgb(194,16,16)' |
| figure.data[1].decreasing.fillcolor = 'rgb(95,141,78)' |
| figure.data[1].decreasing.line.color = 'rgb(40,84,48)' |
|
|
| figure.update_layout(template="plotly_white") |
|
|
| return figure |
|
|
