如何将numpy.array封装为可像内置函数一样调用的array()？

看起来你是想直接调用array()来创建numpy数组，不用每次写numpy.array或者np.array对吧？我来给你拆解解决方法，顺便排查下你遇到的运行异常问题：

1. 最直接的解决方案：把numpy.array导入到当前命名空间 #

不用导入整个numpy库，直接导入array函数本身，这样它就和内置函数一样可以直接调用了：

from numpy import array

# 现在直接用就行，和内置函数没区别
my_array = array([1, 2, 3, 4])
print(my_array)

这种方式会把numpy的array函数直接放到当前代码的全局命名空间里，IDE能识别，运行时也不会找不到函数。

2. 排查运行异常的可能原因 #

你说加了导入后IDE不报错但运行异常，大概率是命名冲突或者导入方式不对：

• 如果你之前写的是import numpy，那直接调用array()肯定会报错，因为这时候array不在当前命名空间，必须写numpy.array()才行；
• 要是你同时导入了其他库的array函数（比如Python标准库的array.array），就会出现命名冲突。比如你同时写了from array import array和from numpy import array，后面的会覆盖前面的，导致代码逻辑和预期不符。

解决冲突的办法是给其中一个函数起别名：

# 给numpy的array起别名，避免和标准库的array冲突
from numpy import array as np_array
from array import array as py_array

# 分别调用各自的array函数
numpy_arr = np_array([1.0, 2.0, 3.0])
python_arr = py_array('f', [1.0, 2.0, 3.0])

3. 如果原代码有自定义封装 #

如果原代码里的array()是某个自定义封装（比如别人写的工具函数，内部调用了numpy.array），那你得先找到这个封装的定义。要是找不到，直接用from numpy import array就能替代它的功能，但要确认原封装有没有额外的逻辑（比如默认参数、数据类型转换），如果有的话，你可能需要自己实现对应的封装函数：

from numpy import array

def my_array(data, dtype='float64'):
    # 模拟原封装的额外逻辑
    return array(data, dtype=dtype)

# 用自定义的my_array替代原代码里的array
result = my_array([1,2,3])

这样调整后，你就能像用内置函数一样调用array()（或者你自定义的封装函数），同时解决运行异常的问题。

内容的提问来源于stack exchange，提问作者Nick X Tsui