[譯] 減小 Python 中循環的使用

• 原文地址：Minimize for loop usage in Python
• 原文做者：Rahul Agarwal
• 譯文出自：掘金翻譯計劃
• 本文永久連接：github.com/xitu/gold-m…
• 譯者：qiuyuezhong
• 校對者：MollyAredtana、shixi-li

減小 Python 中循環的使用

如何以及爲何應該在 Python 中減小循環的使用？

Python 提供給咱們多種編碼方式。

在某種程度上，這至關具備包容性。

來自於任何語言的人均可以編寫 Python。

然而，學習寫一門語言和以最優的方式寫一門語言是兩件不一樣的事情。

在這一系列名爲 Python Shorts 的文章中，我將闡述 Python 提供的一些簡單可是很是有用的結構，一些小技巧以及一些我在數據科學工做中遇到的案例。

在這篇文章中，我將討論 Python 中的 for 循環，以及如何儘可能避免使用它們。

寫 for 循環的 3 種方式：

讓我用一個簡單的例子來解釋下。

假設你想取得一個列表中的平方和。

在機器學習中，當咱們想計算 n 維狀況下兩點之間的距離時，咱們都會面臨這個問題。

你可使用循環很容易的作到這一點。

事實上，我想展現給你我看到的用來完成一樣任務的三種方式，並讓你選擇你認爲最好的方式。

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 的風格。
• 能夠用 List Comprehension 作到嗎？若是是，使用它。
• 能夠不索引數組嗎？若是不是，考慮使用 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

也能夠嘗試使用 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、前端、後端、區塊鏈、產品、設計、人工智能等領域，想要查看更多優質譯文請持續關注 掘金翻譯計劃、官方微博、知乎專欄。