NPTEL Data Science for Engineers Assignment 8 Answer