Nptel | Assignment solution | python for data science | Week 4