Python Array Example Code Plotting Points