[파이썬] 문제 : n개의 숫자 입력 받아서 음수, 양수 별 평균 계

cnt = int(input('입력할 숫자의 갯수를 정해주세요: '))

def find_ave(lst):
    negative_list = []
    positive_list = []
    for n in lst:
        if n < 0:
            negative_list.append(n)
        else:
            positive_list.append(n)
        
    n_avg = sum(negative_list) / len(negative_list)
    p_avg = sum(positive_list) / len(positive_list)
    return negative_list, n_avg, positive_list, p_avg

input_list = []
for i in range(cnt):
    n = int(input(f'{i+1} 번째 숫자: '))
    input_list.append(n)
    
negative_list, n_avg, positive_list, p_avg = find_ave(input_list)
print('양수:',positive_list)
print('양수의 평균:',p_avg)
print()
print('음수:',negative_list)
print('움수의 평균:',n_avg)