os、sys库技巧整理

切换到指定工作目录再执行命令

os.chdir os.system

os.chdir("/home/hadoop/bigdata/hadoop/sbin")
os.system("./start-dfs.sh")

执行Python脚本时传入一个或多个参数

sys.argv

import sys

def main(argv):
    print(argv[1])
    print(argv[2])
    print(argv[3])

if __name__ == "__main__":
    main(sys.argv)

系统中执行python main.py 1 2 3, 将会得到1 2 3的打印。

获取系统环境变量：

os.environ and os.putenv

一、设置系统环境变量

os.environ['环境变量名称']='环境变量值' #其中key和value均为string类型
os.putenv('环境变量名称', '环境变量值')

二、获取系统环境变量

os.environ['环境变量名称'] # 注意，这么获取，如果变量名不存在，则会抛出出异常，可以用os.environ.get('环境变量名称')
os.getenv('环境变量名称') # 该方法尚未尝试过