pandas DataFrame, col
Python
Write a function that takes in as input a dataframe and a column name, and returns the mean for numerical columns and the mode for non-numerical columns. Function Specifications: The function should take two inputs: (df, column_name), where df is a pandas DataFrame, column_name is a str. If the column_name does not exist in df, raise a ValueError. Should return as output the mean if the specified column is numerical and return a list of the mode(s) otherwise. The mean should be rounded to 2 decimal places. If there is more than one mode for a given non-numerical column, the fuction should return a list of all modes. I
def calc_mean_mode(df, column_name): # your code here return
calc_mean_mode(df,'Age')
Expected Outputs: calc_mean_mode(df, 'Age') == 29.7
calc_mean_mode(df, 'Embarked') == ['S']
Trending now
This is a popular solution!
Step by step
Solved in 2 steps with 2 images
