Python Take Average