Data visualization with python assignment