data visualization with python assignment