Python：如何确保函数中使用的模块别名正确？

如何解决Python：如何确保函数中使用的模块别名正确？

我正在尝试使任何人都可以使用函数foo。如何确保它们导入具有特定别名的模块？

#include <iostream>
using namespace std;
int main(int argc,const char * argv[]) {
    int  arr[] = {1,2,3,4,5,6};
    int size = *(&arr + 1) - arr;
    cout << "Number of elements in arr[] is "
         << size;
    return 0;
}

现在我正在做这样的事情，如果他们必须为numpy和pandas使用不同的别名，这似乎是不可持续的。

def foo(object):
    objdtypes = (list,tuple,np.ndarray,pd.core.series.Series)
    if isinstance(object,objdtypes): 
        print("It's there")

解决方法

将包导入函数中，而不是依赖于全局范围。

def foo(obj):
    import pandas as pd
    import numpy as np

    objdtypes = (list,tuple,np.ndarray,pd.core.series.Series)
    if isinstance(obj,objdtypes): 
        print("It's there")

也不要使用object作为变量名，因为那样会遮盖内置object。

当某人导入您的函数时，将使用您使用的别名导入任何必需的模块。例如，您可以创建一个名为mymodule.py的文件，其中包含：

import numpy as np

def myfunc(arg):
    return np.sin(arg)

如果您随后打开Python终端会话并使用：

from mymodule import myfunc

print(myfunc(2))
0.9092974268256817

它会很好地工作。如果您在终端会话中导入具有其他别名的numpy，例如，

import numpy as blah
from mymodule import myfunc

print(myfunc(2))
0.9092974268256817

导入myfunc后，请查看：

myfunc.__globals__
{'__builtins__': {'ArithmeticError': ArithmeticError,...
'__file__': 'module.pyc','__name__': 'module','__package__': None,'myfunc': <function module.myfunc>,'np': <module 'numpy' from '/usr/local/lib/python2.7/dist-packages/numpy/__init__.pyc'>}

您会看到它包含numpy（np）的预期别名，这就是它将要使用的别名。