我们为什么要减少Python中循环的使用

yipeiwu_com5年前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。

避免像传染病一样的循环

从长远来看,你的代码将更具可读性和可维护性。

英文原文地址:Minimize for loop usage in Python

原文作者:Rahul Agarwal

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

python脚本生成caffe train_list.txt的方法

首先给出代码: import os path = "/home/data//" path_exp = os.path.expanduser(path) classes =...

在django模板中实现超链接配置

django中的超链接,在template中可以用{% url 'app_name:url_name' param%} 其中app_name在工程urls中配置的namespace取值,...

Python的bit_length函数来二进制的位数方法

自Python3.1中,整数bit_length方法允许查询二进制的位数或长度。 常规做法: >>> bin(256) '0b100000000' >&g...

Python+tkinter模拟“记住我”自动登录实例代码

Python+tkinter模拟“记住我”自动登录实例代码

本文分享的代码主要是通过Python+tkinter模拟“记住我”自动登录的功能,具体介绍如下。 基本思路:如果某次登录成功,则创建临时文件记录有关信息,每次启动程序时尝试自动获取上次登...

python 字符串常用方法汇总详解

1.字符串大小写转 value = "wangdianchao" # 转换为大写 big_value = value.upper() print(big_value) # 转换为小写...