[python] Use index in pandas to plot data