从if语句中获取平均值Python 3

# 初始化变量
total = 0
count = 0

# 循环输入数字，直到输入0为止
while True:
    num = int(input("请输入一个数字（输入0结束）："))
    
    # 判断是否结束输入
    if num == 0:
        break
    
    # 累加求和
    total += num
    count += 1

# 判断是否有输入数字
if count == 0:
    print("没有输入数字")
else:
    # 计算平均值
    average = total / count
    print("平均值为:", average)