1cateogry_columns=df.select_dtypes(include=['object']).columns.tolist()
2integer_columns=df.select_dtypes(include=['int64','float64']).columns.tolist()
3
4for column in df:
5 if df[column].isnull().any():
6 if(column in cateogry_columns):
7 df[column]=df[column].fillna(df[column].mode()[0])
8 else:
9 df[column]=df[column].fillna(df[column].mean)