3.2.3 循环代码优化技巧

  学过太极拳的朋友应该听过一句话:要把拳练好,必把圈练小。一般而言,功夫越深的人外形动作越小,编写代码也是同样的道理。在满足功能要求的前提下应该追求代码的短小精悍,而这也是对Python内功的考验。虽然不至于“两句三年的,一吟双泪流”,但确实也是需要经过反复推敲的。有不少人编写代码之前没有经过系统地规划,也没有经过深思熟虑,完全是凭着感觉走,想到哪写到哪,写的时候还反复地修改前面的代码。对于一些小的程序这样做是可以的,但是对于大型软件开发,这种“不管黑猫白猫,抓到老鼠就是好猫”的态度是可怕的。编写代码时,不仅要考虑功能,还要考虑性能以及可维护性、可扩展性、可移植性等。

  虽然现在的计算机配置越来越高,内存越来越大,计算速度越来越快,计算能力越来越强,各种云平台更是提供了惊人的存储空间和计算能力,但作为一种美德,还是要合理利用资源,能少用一点资源就省一点,能让代码运行更快一点就尽量优化一下,同时也能给代码进行适当“减肥”来增加可读性。在编写循环语句时,应尽量减少循环内部不必要或无关的计算,将与循环变量无关的代码尽可能提取到循环之外,这样可以提高代码的执行效率。对于使用多重循环嵌套的情况,应尽量较少内层循环中不必要的计算,尽可能的向外提。例如,下面的代码,第二段明显比第一段的运行效率高。

 1 import time
 2 
 3 digits = (1,2,3,4)
 4 start = time.time()
 5 for i in range(1000):
 6     result = []
 7     for i in digits:
 8         for j in digits:
 9             for k in digits:
10                 result.append(i ** 100 + j * 10 + k)
11 
12 print('# 第一段代码执行耗时:{}'.format(time.time() - start))
13 
14 start2 = time.time()
15 for i in range(1000):
16     result = []
17     for i in digits:
18         i *= 100
19         for j in digits:
20             j *= 10
21             for k in digits:
22                 result.append(i + j + k)
23 
24 print('# 第二段代码执行耗时:{}'.format(time.time() - start2))
25 print(result)
26 
27 # 第一段代码执行耗时:0.062400102615356445
28 # 第二段代码执行耗时:0.031200170516967773
29 
30 #[111, 112, 113, 114, 121, 122, 123, 124, 131, 132, 133, 134, 141, 142, 143, 144, 211, 212, 213, 214, 221, 222, 223, 224, 231, 232, 233, 
234, 241, 242, 243, 244, 311, 312, 313, 314, 321, 322, 323, 324, 331, 332, 333, 334, 341, 342, 343, 344, 411, 412, 413, 414, 421, 422,
423, 424, 431, 432, 433, 434, 441, 442, 443, 444]

  另外,在循环中应尽量引用局部变量,局部变量的查询和访问速度比全局变量略快,在使用模块中的方法时,可以通过将其转换为局部变量来提高运行速度。例如下面的代码:

 1 import time
 2 import math
 3 
 4 start = time.time()
 5 for i in range(10000000):
 6     math.sin(i)
 7 
 8 print('# Time Used 1:',time.time() - start)
 9 
10 loc_sin = math.sin
11 start = time.time()
12 for i in range(10000000):
13     loc_sin(i)
14 
15 print('# Time Used 2:',time.time() - start)
16 
17 #我自己的方法
18 loc_sin = math.sin
19 start = time.time()
20 m = map(loc_sin,range(10000000))
21 l = list(m)
22 print('# Time Used 3:',time.time() - start)
23 
24 
25 # Time Used 1: 3.135605573654175
26 # Time Used 2: 2.6520047187805176
27 # Time Used 3: 2.979605197906494

  虽然速度提高并不是非常多,但是对于某些对实时性要求特别高的应用场景一点点的提高也是有意义的。而实际上,上面这段代码还有优化的空间,大家能想到吗?本书其他地方介绍过,尝试着找一下。

  《太极尺寸分毫解》曰“功夫先练开展,后练紧凑”。编写代码也是同样的道理,首先哟啊把代码写对,保证完全符合功能要求,然后再进行必要的优化来提高性能。过早地追求性能优化有时可能会带来灾难而浪费大量精力。代码优化牵涉的面非常广,对程序员的功底要求很高。除了上面介绍的循环代码优化,第2章和第5章中介绍的内容也涉及一些优化的内容。例如,如果经常需要测试一个序列是否包含一个元素应尽量使用字典或集合而不适用列表,连接多个字符串时尽量使用join()方法二不要使用运算符"+",对列表进行元素的插入和删除操作时应尽量从列表尾部进行,等等。

  练习:1 输出“水仙花数”

  所谓水仙花数是指一个3位的十进制数,其各位数字的立方和恰好等于该数本身,例如,153是水仙花数,因为 153 = 1 ** 3 + 5 ** 3 + 3 ** 3

 1 import time
 2 
 3 l = []
 4 
 5 start = time.time()
 6 
 7 for i in range(100,1000):
 8     #获取个位数
 9     g = i % 10
10 
11     #获取十位数
12     s = i // 10 % 10
13 
14     #获取百位数
15     b = i // 100
16 
17     sxh =  g ** 3 + s ** 3 + b ** 3
18     if sxh == i:
19         l.append(sxh)
20 print('# 程序执行耗时:',time.time() - start)
21 print('# 程序执行结果:{}'.format(l))
22 
23 # 程序执行耗时: 0.015599966049194336
24 # 程序执行结果:[153, 370, 371, 407]                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                             

  练习2:打印九九乘法表

 1 for i in range(1,10):
 2     for j in range(1,i + 1):
 3         print(' {} * {} = {}'.format(i,j,i * j),end = ' ')
 4 
 5     print()
 6 
 7 
 8  1 * 1 = 1 
 9  2 * 1 = 2  2 * 2 = 4 
10  3 * 1 = 3  3 * 2 = 6  3 * 3 = 9 
11  4 * 1 = 4  4 * 2 = 8  4 * 3 = 12  4 * 4 = 16 
12  5 * 1 = 5  5 * 2 = 10  5 * 3 = 15  5 * 4 = 20  5 * 5 = 25 
13  6 * 1 = 6  6 * 2 = 12  6 * 3 = 18  6 * 4 = 24  6 * 5 = 30  6 * 6 = 36 
14  7 * 1 = 7  7 * 2 = 14  7 * 3 = 21  7 * 4 = 28  7 * 5 = 35  7 * 6 = 42  7 * 7 = 49 
15  8 * 1 = 8  8 * 2 = 16  8 * 3 = 24  8 * 4 = 32  8 * 5 = 40  8 * 6 = 48  8 * 7 = 56  8 * 8 = 64 
16  9 * 1 = 9  9 * 2 = 18  9 * 3 = 27  9 * 4 = 36  9 * 5 = 45  9 * 6 = 54  9 * 7 = 63  9 * 8 = 72  9 * 9 = 81 

  练习 3:求200以内能被17整除的最大整数

 1 import time
 2 a = 0
 3 
 4 #实现代码 1
 5 start = time.time()
 6 for i in range(1,20000000):
 7     if i % 17 == 0:
 8         a = i
 9 
10 print('# Time Used 1 : {}'.format(time.time() - start))
11 print('# 程序执行结果:{}'.format(a))
12 
13 #实现代码 2
14 start = time.time()
15 for i in range(20000000,1,-1):
16     if i % 17 == 0:
17         a = i
18         break
19 
20 print('# Time Used 2 : {}'.format(time.time() - start))
21 print('# 程序执行结果:{}'.format(a))
22 
23 # Time Used 1 : 3.9176156520843506
24 # 程序执行结果:19999990
25 # Time Used 2 : 0.0
26 # 程序执行结果:19999990
27 
28 
29 #这个耗时差距有点大哦,,还是实现代码 2 比较好啊

  练习 4:自定义一个函数,判断一个数是否为素数

 1 import math
 2 
 3 def is_su(x):
 4     a = 1
 5     for i in range(2,int(math.sqrt(x)) + 1):
 6         if x % i == 0:
 7             a = 2
 8             break
 9 
10     if a == 1:
11         print(x,'是素数')
12     else:
13         print('x = {} 不是素数'.format(x))
14 
15 #函数调用
16 
17 is_su(9)
18 #x = 9 不是素数
19 
20 is_su(7)
21 #7 是素数

  拓展知识:math是用于数学计算的标准库。除了用于平方根函数sqrt()和取整函数ceil(),Python标准库math还提供了最大公约数gcd(),sin()、asin()等三角函数与反三角函数,弧度与角度转换函数degrees() 和 radians(),误差函数erf()、剩余误差函数erfc()、伽马函数gamma(),对函数log()、log2()、log10(),阶乘函数factorial(),常数pi和e等。

  练习 5 鸡兔同笼问题。假设共有鸡、兔30只,脚90只,求鸡、兔各有多少只?

 1 #定义鸡的个数 x
 2 #定义兔的个数 y
 3 
 4 print('# 可能的组合情况如下')
 5 for x in range(0,31):
 6     for y in range(0,31 - x):
 7         if x * 2 + y * 4 == 90 and x + y == 30:
 8             print('# 执行结果,鸡的个数:{},兔的个数:{}'.format(x,y))
 9             
10 # 可能的组合情况如下
11 # 执行结果,鸡的个数:15,兔的个数:15
原文地址:https://www.cnblogs.com/avention/p/8583515.html