@@ -103,7 +103,7 @@ This version is more maintainable code, but it doesn't give the function caller
103103``` {python}
104104def load_filter_and_average(file, grouping_column, ploting_column):
105105 df = pd.read_csv("data/" + file)
106- source = df.groupby(grouping_column).mean().reset_index()
106+ source = df.groupby(grouping_column).mean(numeric_only=True ).reset_index()
107107 chart = alt.Chart(source, width = 500, height = 300).mark_bar().encode(
108108 x=alt.X(grouping_column),
109109 y=alt.Y(ploting_column)
@@ -146,7 +146,7 @@ Another problem with this function is that the means are only printed and not re
146146
147147``` {python}
148148def grouped_means(df, grouping_column):
149- grouped_mean = df.groupby(grouping_column).mean().reset_index()
149+ grouped_mean = df.groupby(grouping_column).mean(numeric_only=True ).reset_index()
150150 return grouped_mean
151151```
152152
@@ -214,7 +214,7 @@ pd.set_option('display.max_columns', 6)
214214``` {python}
215215def load_filter_and_average(file, grouping_column, ploting_column):
216216 df = pd.read_csv("data/" + file)
217- source = df.groupby(grouping_column).mean().reset_index()
217+ source = df.groupby(grouping_column).mean(numeric_only=True ).reset_index()
218218 chart = alt.Chart(source, width = 500, height = 300).mark_bar().encode(
219219 x=alt.X(grouping_column),
220220 y=alt.Y(ploting_column)
@@ -267,7 +267,7 @@ It's best to think of programming functions in the same way as mathematical func
267267
268268``` {python}
269269def grouped_means(df, grouping_column):
270- grouped_mean = df.groupby(grouping_column).mean().reset_index()
270+ grouped_mean = df.groupby(grouping_column).mean(numeric_only=True ).reset_index()
271271 return grouped_mean
272272```
273273
@@ -277,7 +277,7 @@ def grouped_means(df, grouping_column):
277277cereal = pd.read_csv('data/cereal.csv')
278278
279279def bad_grouped_means(grouping_column):
280- grouped_mean = cereal.groupby(grouping_column).mean().reset_index()
280+ grouped_mean = cereal.groupby(grouping_column).mean(numeric_only=True ).reset_index()
281281 return grouped_mean
282282```
283283
0 commit comments