diff --git a/streamlit/part5/basketball_app.py b/streamlit/part5/basketball_app.py index 9dc7283..c0ae5f1 100644 --- a/streamlit/part5/basketball_app.py +++ b/streamlit/part5/basketball_app.py @@ -25,6 +25,8 @@ def load_data(year): raw = df.drop(df[df.Age == 'Age'].index) # Deletes repeating headers in content raw = raw.fillna(0) playerstats = raw.drop(['Rk'], axis=1) + for i in playerstats.filter(regex='%').columns: + playerstats[i] = playerstats[i].astype(float) return playerstats playerstats = load_data(selected_year) @@ -43,6 +45,7 @@ def load_data(year): st.write('Data Dimension: ' + str(df_selected_team.shape[0]) + ' rows and ' + str(df_selected_team.shape[1]) + ' columns.') st.dataframe(df_selected_team) + # Download NBA player stats data # https://discuss.streamlit.io/t/how-to-download-file-in-streamlit/1806 def filedownload(df): @@ -65,4 +68,4 @@ def filedownload(df): with sns.axes_style("white"): f, ax = plt.subplots(figsize=(7, 5)) ax = sns.heatmap(corr, mask=mask, vmax=1, square=True) - st.pyplot() + st.pyplot(f)