Find the Median (python)