Find/count number of students with weight < 50.

def setup():
   name = ["Gun", "Rit", "Frame"]
   ID = [32075, 32078, 32080]
   age = [38, 83, 19]
   weight = [80, 45, 50.2]
   height = [1.74, 1.70, 1.73]
   print(count(weight))
 
def count(w):
   i = 0
   count = 0
   while(i < len(w)):
      if(w[i] < 50):
         count = count + 1
      i = i + 1
   return count

setup()