模块和包

模块

Python会将所有 .py 结尾的文件认定为Python代码文件。

考虑这样一个脚本文件 ex1.py：

%%writefile ex1.py
PI = 3.1416

def sum(lst):
    tot = lst[0]
    for value in lst[1:]:
        tot = tot + value
    return tot

w = [0, 1, 2, 3]
print(sum(w), PI)

Writing ex1.py

执行这个脚本：

!python ex1.py

6 3.1416

为了复用代码，Python可以将这个脚本当作一个模块，使用 import 关键词加载并执行它（这里要求ex1.py在当前工作目录）：

import ex1

6 3.1416

type(ex1)

module

ex1

在导入时，Python会执行一遍模块中的所有内容，所以print()函数的结果被输出了。

ex1.py 中所有的变量都被载入了当前环境中，不过要使用ex1.变量名的方法来查看或者修改这些变量：

ex1.PI

3.1416

ex1.PI = 3.141592653

还可以用ex1.函数名调用模块里面的函数：

ex1.sum([2, 3, 4, 5])

14

为了提高效率，Python只会载入模块一次，已经载入的模块再次载入时，Python并不会真正执行载入操作，哪怕模块的内容已经改变：

import ex1

删除示例文件：

%rm ex1.py

name 属性

有时候想将一个 .py 文件既当作脚本，又能当作模块用，这个时候可以使用 .__name__ 这个属性。

只有当文件被当作脚本执行的时候， __name__的值才会是 '__main__'，所以可以这样定义脚本：

%%writefile ex2.py
PI = 3.1416

def sum(lst):
    """ Sum the values in a list
    """
    tot = 0
    for value in lst:
        tot = tot + value
    return tot

def add(x, y):
    " Add two values."
    a = x + y
    return a

def test():
    w = [0,1,2,3]
    assert(sum(w) == 6)
    print('test passed.')

if __name__ == '__main__':
    test()

Overwriting ex2.py

运行文件：

!python ex2.py

test passed.

当作模块导入， test() 不会执行：

import ex2

可以使用其中的变量：

ex2.PI

3.1416

其他导入方法

import ex2 as e2

e2.PI

3.1416

from ex2 import add, PI

使用 from 后，可以直接使用 add ， PI：

add(2, PI)

5.1416

from ex2 import *

add(3, 4.5)

7.5

这种导入方法不是很提倡，因为如果你不确定导入的都有哪些，可能覆盖一些已有的函数。

删除示例文件：

%rm ex2.py

包

有这样的一个文件夹foo：

foo/

__init__.py
bar.py (defines func)
baz.py (defines zap)
``

在Python中，这样的文件夹 foo 是一个包，可以这样导入其中的内容：

```python
from foo.bar import func
from foo.baz import zap

其中bar 和 baz 都是 foo 文件夹下的 .py 文件。

导入包要求：

• 文件夹 foo 在Python的搜索路径中
• 要有__init__.py，它可以是个空文件。