input: Write a function to calculate the average of a list of numbers in Python.
  
  def avg_list(lst):
      sum = 0
      count = 0
      for i in lst:
          sum += i
          count += 1
      avg = sum / count
      return avg
output: def avg_list(lst):
      return sum(lst) / len(lst)