Python3实现求小于10000的同构数

问题：平方后低位的数恰好等于该数的数是同构数。求小于10000的所有同构数。

编译环境：Python3 + Spyder

一般的代码如下：

for i in range(10000):
   k = str(i * i)                            #平方数
    if(len(k) % 2 == 0):
        m = int(k[(len(k) // 2):len(k)])     #取后位数
        if(m == i):
            print(m,end=" ")
    else:
        m = int(k[((len(k) + 1) // 2) - 1:len(k)])
        if(m == i):
            print(m,end=" ")

使用列表解析后：

from math import ceil
print([i for i in range(1,10000) for n in range(len(str(i * i)),len(str(i * i)) + 1) if i == i * i % (10**(ceil(n / 2))) ])

运算结果：

[1, 5, 6, 25, 76, 376, 625, 9376]