均值正负数个数统计

问题描述
从键盘输入非0整数，以输入0为结束标志。求整数平均值，统计正负数个数。
输入形式
每个整数一行。0为输入结束标志。
输出形式
输出三行。第一行是平均值(average)，保留两位小数。第二行是正数(positive)个数。第三行是负数(negative)个数。
样例输入
1
1
1
0
样例输出
1.00
3
0

def calculates():
    total = 0
    count = 0
    positive_count = 0
    negative_count = 0
    
    while True:
        num = int(input())
        if num == 0:
            break
        total += num
        count += 1
        if num > 0:
            positive_count += 1
        elif num < 0:
            negative_count += 1
    
    average = total / count if count > 0 else 0
    print("{0:.2f}".format(average))
    print(positive_count)
    print(negative_count)

calculates()