Python笔记：import导入本地脚本

导入本地脚本

• import 如果你要导入的 Python 脚本与当前脚本位于同一个目录下，只需输入 import，然后是文件名，无需扩展名 .py。 伪代码如下：

    import useful_functions
    useful_functions.add_five([1, 2, 3, 4])
  

• 我们可以为导入模块添加别名，以使用不同的名称引用它。

    import useful_functions as uf
    uf.add_five([1, 2, 3, 4])
  

• 使用 if main 块

  • 为了避免运行从其他脚本中作为模块导入的脚本中的可执行语句，将这些行包含在 if __name__ == "__main__" 块中。或者，将它们包含在函数 main() 中并在 if main 块中调用该函数。
  • 每当我们运行此类脚本时，Python 实际上会为所有模块设置一个特殊的内置变量 __name__。当我们运行脚本时，Python 会将此模块识别为主程序，并将此模块的 __name__ 变量设为字符串 “main”。对于该脚本中导入的任何模块，这个内置 __name__ 变量会设为该模块的名称。因此，条件 if __name__ == "__main__"会检查该模块是否为主程序。

• 在同一目录下创建如下两个脚本文件，并在终端里运行这些脚本！实验 if main 块并访问导入模块中的对象！

  demo.py 文件

    # demo.py
    import useful_functions as uf
    scores = [88, 92, 79, 93, 85]
    mean = uf.mean(scores)
    curved = uf.add_five(scores)
    mean_c = uf.mean(curved)
    print("Scores:", scores)
    print("Original Mean:", mean, " New Mean:", mean_c)
    print(__name__)
    print(uf.__name__)
  

  useful_functions.py文件

    # useful_functions.py
    def mean(num_list):
        return sum(num_list) / len(num_list)
  
    def add_five(num_list):
        return [n + 5 for n in num_list]
  
    def main():
        print("Testing mean function")
        n_list = [34, 44, 23, 46, 12, 24]
        correct_mean = 30.5
        assert(mean(n_list) == correct_mean)
  
        print("Testing add_five function")
        correct_list = [39, 49, 28, 51, 17, 29]
        assert(add_five(n_list) == correct_list)
  
        print("All tests passed!")
  
    if __name__ == '__main__':
        main()
  

  如果运行 $ python demo.py，那么有如下输出结果：

    Scores: [88, 92, 79, 93, 85]
    Original Mean: 87.4  New Mean: 92.4
    __main__
    useful_functions
  

  如果运行 $ python useful_functions.py，那么有如下输出结果：

    Testing mean function
    Testing add_five function
    All tests passed!