| from tools.ofi_delta import calculate_ofi_and_delta |
| from utils.plots import plot_multiple_columns |
| |
|
|
|
|
| def plot_by_columns(ticker, date, timeframe, session): |
| session.markdown('### Stock Dataframe') |
|
|
| with session.spinner(text="Downloading required data..."): |
| df = calculate_ofi_and_delta(ticker=ticker, date=date, timeframe=timeframe) |
|
|
| session.dataframe(df) |
|
|
| session.markdown('### Stock Dataframe Statistics') |
| session.dataframe(df.describe()) |
|
|
| session.markdown('### Plot By columns') |
|
|
| first_column = session.text_input('Enter the Primary Column Name:') |
| second_column = session.text_input('Enter the Secondary Column Name:') |
| third_column_query = session.selectbox("Do you want to have a third column be plotted on your chart?", ("", "Yes", "No")) |
|
|
| if first_column != '' and second_column != '': |
|
|
| if third_column_query == '' or third_column_query == 'No': |
| plot_figure = plot_multiple_columns(dataframe=df, primary_column=first_column, |
| secondary_column=second_column) |
| session.plotly_chart(plot_figure) |
|
|
| else: |
|
|
| third_column = session.text_input('Enter the ternary Column Name:') |
| scale_query = session.selectbox('Please select the closest column in terms of scale to the third column.', |
| ('', 'first column', 'second column')) |
|
|
| if third_column != '' and scale_query != '': |
| if scale_query == 'first column': |
| plot_figure = plot_multiple_columns(dataframe=df, primary_column=first_column, |
| secondary_column=second_column, ternary_column=third_column, |
| scale_similarity=1) |
| session.plotly_chart(plot_figure) |
|
|
| elif scale_query == 'second column': |
| plot_figure = plot_multiple_columns(dataframe=df, primary_column=first_column, |
| secondary_column=second_column, ternary_column=third_column, |
| scale_similarity=2) |
| session.plotly_chart(plot_figure) |
|
|
| else: |
| session.markdown('### You Have Not Chosen Any Columns Yet!') |
|
|
