Python 基础语法 11 模块

2018-12-03

2.5/2.6 模块使用

模块是Python组织代码的基本方式。一个Python脚本可以单独运行，也可以导入到另一个脚本中运行，当脚本被导入运行时，我们将其称为模块（module）。所有的.py文件都可以作为一个模块导入，因此，模块就是Python脚本。

使用sys.path查看默认模块的路径

In [122]: import sys

In [123]: sys.path
Out[123]:
['',
'/usr/bin',
'/usr/lib64/python27.zip',
'/usr/lib64/python2.7',
'/usr/lib64/python2.7/plat-linux2',
'/usr/lib64/python2.7/lib-tk',
'/usr/lib64/python2.7/lib-old',
'/usr/lib64/python2.7/lib-dynload',
'/usr/lib64/python2.7/site-packages',
'/usr/lib/python2.7/site-packages',
'/usr/lib/python2.7/site-packages/IPython/extensions',
'/root/.ipython']

插入自定义模块路径

1）使用sys.path.append

1 2	mkdir /home/python In [124]: sys.path.append('/home/python')

2）在profile或者.bashrc更改系统环境变量

1
2
3

vim /etc/profile
或者
vim .bashrc

添加下面参数

export PYTHONPATH=/home/python

 echo $PYTHONPATH
 /home/python

练习

自建一个和wc一样功能的模块

cd /home/python/
vim wc.py

#!/usr/bin/python

def wordCount(s):
    chars = len(s)  ##统计字符
    words = len(s.split())  ##统计字数
    lines = s.count('\n')  ##统计行数
    print lines,words,chars

s = open('/etc/passwd').read()  ##打开文件
wordCount(s)  ##使用函数

创建另一个文件来插入wc模块

vim 1.py

#!/usr/bin/python

import wc

python 1.py  ##执行1.py
26 42 1230

优化代码为了让其他脚本插入wc模块不执行open语句而wc自己单独执行时才执行open语句可以使用__name__ == '__main__' 它表示脚本自己执行自己，但脚本被当作模块插入就是等于脚本名字,这里是__name__ == wc.py

#!/usr/bin/python

def wordCount(s):
    chars = len(s)
    words = len(s.split())
    lines = s.count('\n')
    print lines,words,chars

if __name__ == '__main__':  ##wc脚本自己执行自己就会执行下面语句
    s = open('/etc/passwd').read()
    wordCount(s)

python wc.py  
26 42 1230

#!/usr/bin/python

import wc

s = 'hello world \n aaaaaa '

wc.wordCount(s)  ##其他脚本插入wc模块调用wordCount方法
 
python 1.py
1 3 21

Python的模块可以按目录组织为包

创建一个包的步骤：
1）创建一个名字为包名的目录
2）在该目录下创建一个__init__.py文件（说明只是个包）
3）根据需要，在该目录下存放脚本文件或已编译的扩展及子包

import 包名.模块名 或者 from 包名 import 模块名

1
2
3

mkdir bao  ##创建目录作为包
touch __init__.py  ##创建init文件
mv 1.py wc.* bao/  把脚本放进包里

from 包名 import 模块名

In [1]: from bao import wc

In [2]: wc.wordCount('asdf')
0 1 4

from 包名.模块名 import 函数名 [as别名]

In [5]: from bao.wc import wordCount

In [6]: wordCount('asdf')
0 1 4

In [7]: from bao.wc import wordCount as wc  ##帮函数wordCount取别名wc

In [8]: wc('asdf')
0 1 4

import 包名.模块名

In [3]: import bao.wc

In [4]: bao.wc.wordCount('asdf')
0 1 4