z=int(input("Введите количество цифр пароля(1-2): " ))
for i in range(0,10):
for e in range(0,10):
if z==1:
print(i)
break
elif z==2:
print(i,e)
Объяснение:
В if надо было поставить оператор break. Программа будет завершать вложенный цикл после вывода одного значения