Python实现阿姆斯特朗数时代码有错误，请问在此基础上如何改？

Question

请在此基础上改，如果改完以后您有更好的方法请一起发过来，谢谢您！

'''
如果一个n位正整数等于其各位数字的n次方之和,则称该数为阿姆斯特朗数。 例如1^3 + 5^3 + 3^3 = 153。
1000以内的阿姆斯特朗数： 1, 2, 3, 4, 5, 6, 7, 8, 9, 153, 370, 371, 407。
检测用户输入的数字是否为阿姆斯特朗数
'''
n = input('请输入数字：')
length = int(len(n))
for i in range(1001):
    li = list(n)
    for j in li:
        list1 = int(j) ** length
    if sum(list1) == n:
        print('True')
    else:
        print('False')

Answer 1

#利用列表解析式，先求出从1到1000所有阿数，然后对输入数进行判断。
inp = input('请输入1000以内的一个数字：')
ams = []
for i in range(1, 1001):
    leng = len(str(i))
    s = [int(x)**leng for x in str(i)]
    if sum(s) == i:
        ams.append(i)

if int(inp) in ams:
    print('True')
else:
    print('False')