Импорт по всему миру

Question

Я создал модуль, который будет использоваться в нескольких сценариях python. Структура выглядит следующим образом:

Главный файл:

import numpy as np 
from mymodule import newfunction 
f = np.arange(100,200,1) 
a = np.zeros(np.shape(f)) 
c = newfunction(f) 

mymodule.py:

def newfunction(f): 
    import numpy as np 
    b = np.zeros(np.shape(f)) 
    return b 

if __name__ == "__main__": 
    import numpy as np 

Не возражаете функциональность этой программы, но проблема в том, что, когда я его запускаю , Я получаю «NameError: глобальное имя« нули »не определено».

Что я пропустил здесь?

Answer 1

mymodule.py не видит:

import numpy as np 

заявление (ы). Операция «import» в Python не работает как #include в C++, она просто создает новый словарь объектов, содержащихся в импортированном модуле. Если вы хотите использовать идентификатор 'np' в этом словаре, вы должны явно импортировать его там.

Что касается

if __name__ == "__main__": 
    import numpy as np 

- это только называется, когда вы выполняете mymodule.py как отдельный скрипт, который, вероятно, это не так в этом вопросе.

EDIT:

OP изменил пример кода, добавив import numpy as np в его функции, и мой ответ на первоначальный вопрос.