从子字典导入函数

Question

我喜欢从另一个字典 python 文件中导入函数。例如，

A.py包含一个get_total()函数，如果用户'from A import get_total'，B.py可以得到这个函数。

但是，当 A.py 需要导入 numpy 模块并在 get_total() 函数中使用 numpy.sum() 时，B.py 无法使用 A 中的 get_total() 函数。导致错误 "NameError: name 'np'未定义”，发生了什么？

A.py：

import numpy as np
def get_total(list):
    return np.sum(list)

B.py：

from A import get_total

a = [1,2,3,4,5]

b=get_total(a)

print(b)

错误：

      1 import numpy
----> 2 def get_total(list):
      3         return numpy.sum(list)
      4 
      5 c = [1,2,3,4,5]

NameError: name 'np' is not defined

Answer 1

您还需要将 numpy 导入 B.py：

import numpy as np
from A import get_total
a = [1,2,3,4,5]
b=get_total(a)
print(b)

编辑：这不是必需的，因为所有 numpy 计算都发生在 A.py 中，它已导入 numpy。