仅被main函数使用的import语句应放在main函数顶部还是if __name__ == "__main__"内？

结论非常明确：这类仅在脚本直接运行时才会用到的导入语句，优先放在if __name__ == "__main__"代码块的顶部；如果你把执行逻辑封装成了独立的def main():函数，也可以放在main函数内部的顶部，两种写法都符合Python社区的通用规范。

这么做的核心原因 #

• 避免不必要的导入开销：放在模块最顶部的import会在模块被其他脚本导入时直接执行，如果依赖仅在运行main逻辑时才需要，会让其他导入该模块的场景平白增加不必要的加载耗时，也会让模块的依赖列表看起来比实际更冗余。
• 避免可选依赖导致的导入错误：如果这类仅main使用的依赖是可选安装的，放在模块顶部会导致没装对应依赖的用户根本无法导入该模块的其他通用功能，完全不符合模块封装的预期。
• 逻辑更清晰：把仅属于运行时的依赖和模块通用依赖分开存放，其他人读代码的时候能快速区分哪些是模块本身的依赖，哪些是仅直接运行脚本才需要的依赖。

写法示例 #

直接写在if __name__ == "__main__"块的情况 #

# 模块通用依赖，统一放在文件最顶部
import os
import sys

def common_util():
    """供其他脚本导入使用的通用工具函数"""
    return os.path.abspath(".")

if __name__ == "__main__":
    # 仅直接运行脚本时需要的依赖，放在块顶部
    import argparse
    import matplotlib.pyplot as plt

    # 执行逻辑
    parser = argparse.ArgumentParser(description="生成统计图表")
    parser.add_argument("input_path", type=str)
    args = parser.parse_args()
    # 其余运行逻辑

封装为main函数的情况 #

# 模块通用依赖，统一放在文件最顶部
import os
import sys

def common_util():
    """供其他脚本导入使用的通用工具函数"""
    return os.path.abspath(".")

def main():
    # 仅main函数使用的依赖，放在函数内部顶部
    import argparse
    import matplotlib.pyplot as plt

    parser = argparse.ArgumentParser(description="生成统计图表")
    parser.add_argument("input_path", type=str)
    args = parser.parse_args()
    # 其余运行逻辑

if __name__ == "__main__":
    main()

内容的提问来源于stack exchange，提问作者beardeadclown