code for visualizing data with python