10條Python代碼優化技巧

Python是一種功能強大的解釋型編程語言。我們可以通過下面的10條優化技巧來減少代碼量並提高代碼的運行效率。

1. 善用列表（list）推導式

問題：輸出1萬內每個數的平方。 發現用列表推導式所用時間更少。

from timeit import default_timer as timer

start = timer()
square =[]
for n in range(10000):
    square.append(pow(n,2))
end = timer()
print("優化前耗時：",end - start)

#優化後
start2=timer()
square=[pow(n,2) for n in range(10000)]
end2=timer()
print("優化後耗時：",end2 - start2)
print(square)

2. 用Python生成器節約運行內存

如果對1萬內每個數的平方求和，換用Python生成器來存儲過程數據，我們發現占用的內存更少。

import sys
square = [n**2 for n in range(10000)]  
print(sum(square))
print(sys.getsizeof(square),'字節') #優化後
square = (n\*\*2 for n in range(10000))
print(sum(square))
print(sys.getsizeof(square),'字節')

3. 用sorted()函數快速實現序列元素的升序排列，用reverse=True實現降序

lst=[20, 40, 20, 20, 40, 60, 70]
print(sorted(list(lst),reverse=True))

4. 用集合set快速過濾掉重復元素

lst=[20, 40, 20, 20, 40, 60, 70]
# print(sorted(list(lst),reverse=True))
print(set(lst))

5. 用enumerate遍歷輸出序列結構中元素的位置和值 用enumerate後，輸出的結果不變，但代碼更簡潔。

dogs = ['Harley', 'Phantom', 'Lucky', 'Dingo']
count = 1
for name in dogs:
    print(count, name)
    count += 1
#優化後
print('優化後')
for i,name in enumerate(dogs,start=1):
    print(i,name)

6. 用字典dictionary中get() 和setdefault()設置默認鍵的值

roman_nums = {'I':1, 'II':2, 'III':3, 'IV':4 }
# value=roman_nums['V']  # 訪問不存在的鍵值會報錯 KeyError: 'V'
value=roman_nums.get('V',5)
print('value=',value)
print(roman_nums)
value = roman_nums.setdefault('V',5)
print(roman_nums)

7. 用Counter統計序列結構中出現次數最多的元素

#統計出現次數最多的元素
#首先引入該方法
from collections import Counter
s='hello'
print(Counter(s)) 

#法二
print([(x,s.count(x)) for x in set(s)])

8. f-Strings (Python 3.6+)格式化字符串

name,major = "world","computer science"

s="Hello, %s. I major in %s." % (name,major)
s1="Hello, {}. I major in {}.".format(name, major)
s2=f"Hello, {name}. I major in {major}." #推薦用法
print(s,s1,s2,sep='\n')

9. join()實現子字符串的拼接

s=['H','e','l','l','o']
s1=''
for x in s:
    s1+=x
print(s1)

s1=''.join(s)#推薦用法
print(s1)

10. 字典合並的N種方法

#法一
x = {'a': 1, 'b': 2}
y = {'b': 10, 'c': 11}
x.update(y)
print(x)
#法二 Python3.9+
z = x | y
print(z)
#法三 Python3.5+
z = {**x, **y}
print(z)
#法四
z = dict(list(x.items()) + list(y.items()))
print(z)

如果喜歡，不忘了在微信文章的下面一鍵三連（分享，點贊和收藏）。關注微信公眾號“樂享Python”。