Visualization and preparation in pandas