Find average of n elements

 def forloop():

    n = int(input("Enter no. of elements to be inserted: "))

    a = []

    for i in range(0,n):

        elem = float(input("Enter Element: " ))

        a.append(elem)

    avg = sum(a)/n

    print("Average of elements: ",avg)

# while loop

def whileloop():

    newlist = list()

    while True:

        elem = input("Enter elem: ")

        if elem=="done":

            break

        value = float(elem)

        newlist.append(value)

    avg = sum(newlist)/len(newlist)

    print("Average of Elements: ",avg)

forloop()

whileloop()