- 原文地址:Minimize for loop usage in Python
- 原文做者:Rahul Agarwal
- 譯文出自:掘金翻譯計劃
- 本文永久連接:github.com/xitu/gold-m…
- 譯者:qiuyuezhong
- 校對者:MollyAredtana、shixi-li
如何以及爲何應該在 Python 中減小循環的使用?前端
Python 提供給咱們多種編碼方式。python
在某種程度上,這至關具備包容性。android
來自於任何語言的人均可以編寫 Python。ios
然而,學習寫一門語言和以最優的方式寫一門語言是兩件不一樣的事情。git
在這一系列名爲 Python Shorts 的文章中,我將闡述 Python 提供的一些簡單可是很是有用的結構,一些小技巧以及一些我在數據科學工做中遇到的案例。github
在這篇文章中,我將討論 Python 中的 for
循環,以及如何儘可能避免使用它們。後端
讓我用一個簡單的例子來解釋下。數組
假設你想取得一個列表中的平方和。機器學習
在機器學習中,當咱們想計算 n 維狀況下兩點之間的距離時,咱們都會面臨這個問題。oop
你可使用循環很容易的作到這一點。
事實上,我想展現給你我看到的用來完成一樣任務的三種方式,並讓你選擇你認爲最好的方式。
x = [1,3,5,7,9]
sum_squared = 0
for i in range(len(x)):
sum_squared+=x[i]**2
複製代碼
當我在 Python 代碼中看到以上代碼的時候,我知道這我的是擁有 C 或者 Java 背景的。
完成一樣的事情,更 Pythonic 的方式是:
x = [1,3,5,7,9]
sum_squared = 0
for y in x:
sum_squared+=y**2
複製代碼
這樣更好了。
我沒有索引這個列表。而且個人代碼更具備可讀性。
可是,更 Pythonic 的方式一行就能夠完成。
x = [1,3,5,7,9]
sum_squared = sum([y**2 for y in x])
複製代碼
這種方法稱爲 List Comprehension,這極可能是我愛上 Python 的緣由之一。
你也能夠在 List Comprehension 中使用 if
。
假設咱們只想要偶數的平方數列表。
x = [1,2,3,4,5,6,7,8,9]
even_squared = [y**2 for y in x if y%2==0]
# 輸出結果:
[4,16,36,64]
複製代碼
if-else
?
若是咱們同時想要偶數的平方數和奇數的立方數呢?
x = [1,2,3,4,5,6,7,8,9]
squared_cubed = [y**2 if y%2==0 else y**3 for y in x]
# 輸出結果:
[1, 4, 27, 16, 125, 36, 343, 64, 729]
複製代碼
太棒了!
所以,大致上遵循這個具體的準則:每當你想寫一個 for
語句的時候,你應該問本身如下的問題,
for
作到嗎?更 Pythonic 的風格。enumerate
。什麼是 enumerate
?
有時咱們既須要數組中的索引,也須要數組中的值。
在這種狀況下,我更喜歡使用 enumerate 而不是索引列表。
L = ['blue', 'yellow', 'orange']
for i, val in enumerate(L):
print("index is %d and value is %s" % (i, val))
# 輸出結果:
index is 0 and value is blue
index is 1 and value is yellow
index is 2 and value is orange
複製代碼
有個規則是:
毫不索引一個列表,若是你能不使用它。
也能夠嘗試使用 Dictionary Comprehension,它是 Python 中相對較新的補充,語法和 List Comprehension 很類似。
讓我用一個例子來解釋。我想爲 x 中的每一個值獲取一個 dictionary(key:平方值)。
x = [1,2,3,4,5,6,7,8,9]
{k:k**2 for k in x}
# 輸出結果:
{1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}
複製代碼
若是隻想獲得偶數值的 dictionary 怎麼辦?
x = [1,2,3,4,5,6,7,8,9]
{k:k**2 for k in x if x%2==0}
# 輸出結果:
{2: 4, 4: 16, 6: 36, 8: 64}
複製代碼
若是想同時獲得偶數值的平方和奇數值的立方怎麼辦?
x = [1,2,3,4,5,6,7,8,9]
{k:k**2 if k%2==0 else k**3 for k in x}
# 輸出結果:
{1: 1, 2: 4, 3: 27, 4: 16, 5: 125, 6: 36, 7: 343, 8: 64, 9: 729}
複製代碼
最後,我要說的是,雖然看上去很容易將從其餘語言得到的知識移用到 Python 上,但若是繼續這樣作,你將沒法理解到 Python 的優美。當咱們用 Python 的方式使用它,它的功能要強大得多,也要有趣得多。
因此,當須要 for
循環的時候,使用 List Comprehensions 和 Dictionary Comprehensions。當須要數組索引的時候,使用 enumerate
。
避免像傳染病同樣的循環
從長遠來看,你的代碼將更具可讀性和可維護性。
另外,若是您想了解更多關於 Python 3 的知識,我想推薦密歇根大學的一門優秀課程 Intermediate level Python。必定要去看看。
未來我也會寫更多適合初學者的文章。請讓我知道你對這個系列的見解。關注個人 Medium 或訂閱個人博客以瞭解相關信息。
和往常同樣,我歡迎反饋和建設性的評論,能夠經過 Twitter 聯繫 @mlwhiz。
最初在 2019 年 4 月 23 號發佈於 mlwhiz.com。
若是發現譯文存在錯誤或其餘須要改進的地方,歡迎到 掘金翻譯計劃 對譯文進行修改並 PR,也可得到相應獎勵積分。文章開頭的 本文永久連接 即爲本文在 GitHub 上的 MarkDown 連接。
掘金翻譯計劃 是一個翻譯優質互聯網技術文章的社區,文章來源爲 掘金 上的英文分享文章。內容覆蓋 Android、iOS、前端、後端、區塊鏈、產品、設計、人工智能等領域,想要查看更多優質譯文請持續關注 掘金翻譯計劃、官方微博、知乎專欄。