python程序编写

求出所有平方数。3025这个数具有一种独特的性质：将它平分为二段，即30和25，使之相加后求平方，即(30+25)^2，恰好等于3025本身。请编写程序求出具有这样性质的全部4位数。

res=[]
for i in range(1000,10000):
    if (int(str(i)[:2])+int(str(i)[2:]))**2==i:
        res.append(i)
print(res)

输出：

[2025, 3025, 9801]