Average number calculation in python