Expression error in 02-Working-With-Data/07-python/notebook.ipynb

[aniveshkanodia]

Describe the bug
Code to demonstrate the concept of grouping in pandas errors out because it tries to perform .mean() to text column B

I believe it should be - df.groupby(by='LenB')[['A','DivA']].mean()

[Vednawale]

import pandas as pd

Sample DataFrame

data = {
'LenB': ['short', 'long', 'short', 'long'],
'A': [1, 2, 3, 4],
'DivA': [10, 20, 30, 40],
'B': ['text1', 'text2', 'text3', 'text4'] # Non-numeric column
}

df = pd.DataFrame(data)

Group by 'LenB' and calculate mean of only numeric columns

result = df.groupby(by='LenB')[['A', 'DivA']].mean()
print(result)