在 while 循环中计算用户的平均值（通过提示输入数据），直到 Python 中的某个条件

Question

I have written the following code which is working fine for me and I want to write the same code in a function (ie, def function_name()) - which I have tried below as well but that's not a good approach and it's poorly written). So, how can I fix the code and how can I put input in an argument of my function name upon invoking/calling argument so that I can get my ultimate answer.

# Normal code:
from statistics import mean 
lst = list()
while True:
    user_int = input("Enter a number: ")
    if user_int == "done":
        break
    try:
        float_user_int = float(user_int)
    except:
        print("Bad input. Enter a valid number")
        continue 
    lst.append(float_user_int)
print(mean(lst)) 

# Function Method I have tried which is working fine but it's poorly written and not yielding correct output. Rather, I am getting an error i.e., NameError: name 'float_user_input' is not defined. How can I write user_input integer numbers in an argument of my function name?

def avg_list(float_user_input):
    from statistics import mean # import statistics
    lst = list()
    while True:
        user_int = input("Enter a number: ")
        if user_int == "done":
            break
        try:
            float_user_int = float(user_int)
        except:
            print("Bad input. Enter a valid number")
            continue # quit()
        lst.append(float_user_int)
    return(mean(lst)) # statistics.mean()
print(avg_list(float_user_input))

# This is the one I tried and it seems to be working with function but I don't think this code is perfect because I have done nothing in it except copy pasting the entire code under def function statement. How can I put user input number in an argument of function name?

def avg_list():
    from statistics import mean # import statistics
    lst = list()
    while True:
        user_int = input("Enter a number: ")
        if user_int == "done":
            break
        try:
            float_user_int = float(user_int)
        except:
            print("Bad input. Enter a valid number")
            continue # quit()
        lst.append(float_user_int)
    return(mean(lst)) # statistics.mean()
print(avg_list())

Your help would be highly appreciated. Thanks.

Answer 1

Remove float_user_input from function declaration and function call

Answer 2

I think you want a function which takes input list from user such as get_input_list and calculate the mean as below:

from statistics import mean # import statistics

def avg_list(input_list):
    return(mean(input_list))


def get_input_list():
    # Take user input here
    lst = list()
    while True:
        user_int = input("Enter a number: ")
        if user_int == "done":
            break
        try:
            float_user_int = float(user_int)
        except:
            print("Bad input. Enter a valid number")
            continue
        lst.append(float_user_int)
    return lst

# call function
print(avg_list(get_input_list()))