Python中能否通过导入外部脚本函数为当前脚本导入库？及cv2未定义问题排查

这是个挺巧妙的思路，但你遇到的报错核心原因是Python的变量作用域规则：当你在load_modules()函数内部执行import cv2时，cv2这个模块对象只存在于该函数的局部命名空间里。函数执行完毕后，局部命名空间里的内容外部无法访问，所以主脚本的全局作用域根本找不到cv2这个名字，自然会抛出未定义的错误。

下面给你几种可行的解决方案，你可以根据实际需求选择：

方案1：让函数返回所有需要的模块 #

修改setup.py中的load_modules()函数，让它把导入的模块作为返回值传递出去：

def load_modules():
    import cv2
    import numpy as np
    import pandas as pd
    from scipy import ndimage
    import os
    from matplotlib import pyplot as plt
    import matplotlib.image as mpimg
    import time
    import math
    import random
    from scipy.ndimage.interpolation import map_coordinates
    from scipy.ndimage.filters import gaussian_filter
    
    # 将所有模块打包返回
    return (cv2, np, pd, ndimage, os, plt, mpimg, time, math, random, 
            map_coordinates, gaussian_filter)

然后在主脚本中按顺序接收这些返回值：

from setup import load_modules

# 按函数返回的顺序接收模块
cv2, np, pd, ndimage, os, plt, mpimg, time, math, random, map_coordinates, gaussian_filter = load_modules()

img = cv2.imread("image.jpg", 1)

这种方式清晰明了，能让阅读代码的人一眼看到主脚本依赖的所有模块。

方案2：在函数中将模块注入全局作用域 #

如果你不想手动接收返回值，可以在load_modules()里用globals()函数，把模块直接添加到主脚本的全局命名空间中：
修改setup.py：

def load_modules():
    # 声明要操作全局命名空间
    global cv2, np, pd, ndimage, os, plt, mpimg, time, math, random, map_coordinates, gaussian_filter
    
    import cv2
    import numpy as np
    import pandas as pd
    from scipy import ndimage
    import os
    from matplotlib import pyplot as plt
    import matplotlib.image as mpimg
    import time
    import math
    import random
    from scipy.ndimage.interpolation import map_coordinates
    from scipy.ndimage.filters import gaussian_filter

主脚本里只需要简单调用函数即可：

from setup import load_modules
load_modules()

img = cv2.imread("image.jpg", 1)

这种方式虽然省事，但会让代码的依赖关系变得隐蔽，其他人看主脚本时会疑惑cv2是从哪来的，长期来看不利于代码维护。

方案3：最规范的做法——直接在模块中导出依赖 #

其实更符合Python编码风格的做法是，把setup.py当成一个依赖导出模块，直接在模块顶层完成所有库的导入：
setup.py内容改为：

import cv2
import numpy as np
import pandas as pd
from scipy import ndimage
import os
from matplotlib import pyplot as plt
import matplotlib.image as mpimg
import time
import math
import random
from scipy.ndimage.interpolation import map_coordinates
from scipy.ndimage.filters import gaussian_filter

然后主脚本直接从setup导入需要的模块：

from setup import cv2, np, pd, ndimage, os, plt, mpimg, time, math, random, map_coordinates, gaussian_filter

img = cv2.imread("image.jpg", 1)

如果图省事也可以用from setup import *，但不推荐这种写法——它会让命名空间变得混乱，还是明确导入需要的模块更清晰。

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