<filename>teste.py
res = input()
if(res == "61"):
    print ('Brasilia')
elif(res == "71"):
    print ('Salvador')
elif (res == "11"):
    print ('Sao Paulo')
elif (res == "21"):
    print ('Rio de Janeiro')
elif (res == "32"):
    print ('Juiz de Fora')
elif (res == "19"):
    print ('Campinas')
elif (res == "27"):
    print ('Vitoria')
elif (res == "31"):
    print ('Belo Horizonte')
else:
    print('DDD nao cadastrado')
<filename>teste.py
res = input()
if(res == "61"):
    print ('Brasilia')
elif(res == "71"):
    print ('Salvador')
elif (res == "11"):
    print ('Sao Paulo')
elif (res == "21"):
    print ('Rio de Janeiro')
elif (res == "32"):
    print ('Juiz de Fora')
elif (res == "19"):
    print ('Campinas')
elif (res == "27"):
    print ('Vitoria')
elif (res == "31"):
    print ('Belo Horizonte')
else:
    print('DDD nao cadastrado')