【python】timeit模块方法的学习以及__main__ 的含义

python的timeit模块可以对代码执行效率进行分析。

class timeit.Timer(stmt=‘pass’, setup=‘pass’, timer=<timer function>)
Timer是测量小段代码执行速度的类。

• stmt参数是要测试的代码语句（statment）；
• setup参数是运行代码时需要的设置；
• timer参数是一个定时器函数，与平台有关。

timeit.Timer.timeit(number=1000000)
Timer类中测试语句执行速度的对象方法。number参数是测试代码时的测试次数，默认为1000000次。方法返回执行代码的平均耗时，一个float类型的秒数。

def test1():
   l = []
   for i in range(1000):
      l = l + [i]
def test2():
   l = []
   for i in range(1000):
      l.append(i)
def test3():
   l = [i for i in range(1000)]
def test4():
   l = list(range(1000))

from timeit import Timer

t1 = Timer("test1()", "from __main__ import test1")
print("concat ",t1.timeit(number=1000), "seconds")
t2 = Timer("test2()", "from __main__ import test2")
print("append ",t2.timeit(number=1000), "seconds")
t3 = Timer("test3()", "from __main__ import test3")
print("comprehension ",t3.timeit(number=1000), "seconds")
t4 = Timer("test4()", "from __main__ import test4")
print("list range ",t4.timeit(number=1000), "seconds")

# ('concat ', 1.7890608310699463, 'seconds')
# ('append ', 0.13796091079711914, 'seconds')
# ('comprehension ', 0.05671119689941406, 'seconds')
# ('list range ', 0.014147043228149414, 'seconds')

但是并没有搞懂第二个参数的含义。
何谓__main__呢？

python官方文档v3.7.4给出如下解释：

根据官方文档，我们知道__main__即代指当前文件（module?）。
只要你创建了一个模块（一个.py文件），这个模块就有一个内置属性name生成，该模块的 name 的值取决于如何使用这个模块。如果直接运行该模块，那么__name__ == “__main__”；如果你把这个文件 import到了另外一个模块，__name__的值就不再是__main__。

做如下测试：

# print_main_function.py
print('Hello World!')
def main():
    print('This message is from main function.')

if __name__ == '__main__':
    main()

在print_main_function.py中首先直接打印一条语句。再通过条件控制，打印main函数。

# print_main_function.py
import print_main_function

print("This message is from main module")

在print_main_module.py中import print_main_function模块，然后直接打印一条语句。

结果如下所示（顺序对应）：

两段代码都打印了hello world。这一条语句在PMF（print_main_function）文件之中是不加控制的打印，因此在被包含到别的文件后也会打印出来。而PMF文件通过__name__进行控制的main函数，在被导入新的文件后，判断语句无法通过。由此可以证明，__main__的作用域即为python文件自身所在的域。