Написать программу, которая будет при вводе с клавиатуры заданного количества чисел подсчитывать сколько из них больше нуля, сколько из них меньше нуля в python

P,n=0,0
for i in range(int(input())):
a=int(input())
if (a>0):
p+=1
if (a<0):
n+=1
print("positive: {}\nnegative: {}".format(p,n))