Skip to content

Commit 8ec8bb2

Browse files
committed
fix issues caused by pandas
1 parent a0c1b6d commit 8ec8bb2

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

modules/module6/slides/module6_22.qmd

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ This version is more maintainable code, but it doesn't give the function caller
103103
```{python}
104104
def 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}
148148
def 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}
215215
def 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}
269269
def 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):
277277
cereal = pd.read_csv('data/cereal.csv')
278278
279279
def 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

Comments
 (0)