[Python] 肇事車輛-找車牌

有一車禍發生在某個城市的十字路口

目擊者 a b c  三人

a 說，該車牌前兩個號碼一樣

b 說，該車牌後兩個號碼一樣

c 說，該車牌是某個數的平方

問，該車牌有幾種可能?

以下是想法

1000的平方根是30左右

9999的平方根是100左右

所以設定一個loop，讓 i 從30跑到100，對每個位數取出四位，abcd，

a == b
且
c == d
那個 i 就是答案

程式碼

def takeEachDig(number):
    """
    這個回傳的list，
    """
    percentile = []
    percentile = list(number)
    percentile.reverse()
    return(percentile)

#test = input("test input: ")

answer = []

for i in range(33,100):
    i = i **2
    temp = takeEachDig(str(i))
    if temp[0] == temp [1] and temp[2] == temp[3]:
        answer.append(i)
    else:
        pass
    

print(answer)