Python Array Example Code Plotting