python 學習總結4

標籤：改進   編寫   value   題目   []   class   學習總結   new   div   

今天在codewears上做了一道題，僅為6kyu層級，但自己收穫良多

題目為：編寫一個函數，sum_dig_pow(a, b)，找到range [a, b]之間所有符合特殊規則的的數字，將其放入列表中，並返回。

這種特殊規則，如下所示;

135 = 1^1 + 3^2 + 5^3

89 = 8^1 + 9^2

自己最開始寫的函數為：

import mathdef sum_dig_pow(a, b):     lis=[]    for i in range(a,b+1):        li=[]        new_i=i        #找到數字是幾位元        k=int(math.log10(new_i))        #將數字切分，如將135切分為[1,3,5]        for j in range(k+1):            li.append(int(new_i/(math.pow(10,k-j))))            new_i=new_i%(math.pow(10,k-j))                #求如 1^1 + 3^2 + 5^3所示的和值        ex=1        sum=0        for x in li:            sum+=math.pow(x,ex)            ex+=1                #判斷和值是否與當前數字相等        if sum==i:            lis.append(i)                             return lis     

　　之後複習了之前的一些內容，對上述代碼進行改進；

lis=[]def sum_dig_pow(a, b):     for i in range(a,b+1):        #將數字切分字串，如將135切分為[‘1‘,‘3‘,‘5‘]        li=re.findall(‘[0-9]‘,str(i))        sum=0                #求和        for key,value in enumerate(li):            sum+=pow(int(value),int(key)+1)                        #判斷            if sum==i:                lis.append(i)    return lis   

　　

python 學習總結4