Visualization of experimental data in python