如何创建自定义NumPy函数？及解决NumPy版本查询报错问题

一、如何创建自定义NumPy函数 #

• 直接编写基于NumPy数组的函数
  利用NumPy的向量化特性，函数可直接接收NumPy数组作为输入，内部用NumPy内置操作实现逻辑，自动支持批量处理。示例：

  import numpy as np
  
  def custom_square(arr):
      return arr ** 2
  
  test_arr = np.array([1,2,3,4])
  print(custom_square(test_arr))  # 输出 [ 1  4  9 16]
  

• 使用np.vectorize包装普通函数
  若函数原本仅处理单个元素，可通过np.vectorize将其转换为支持数组输入的函数（注意：此方法本质是循环，性能不如原生NumPy操作，适合简单场景）。示例：

  def single_element_func(x):
      return x * 2 + 1
  
  vectorized_func = np.vectorize(single_element_func)
  test_arr = np.array([1,2,3])
  print(vectorized_func(test_arr))  # 输出 [3 5 7]
  

• 使用np.frompyfunc创建通用函数
  这是更底层的实现方式，指定输入输出数量后，可将普通函数转换为NumPy通用函数（ufunc）。示例：

  def add_one(x):
      return x + 1
  
  numpy_add_one = np.frompyfunc(add_one, 1, 1)
  test_arr = np.array([0,1,2])
  print(numpy_add_one(test_arr))  # 输出 [1 2 3]
  

二、解决AttributeError: module 'numpy' has no attribute '__version__'报错 #

从你的代码和输出来看，问题源于两点：

1. 语法错误：获取NumPy版本的正确写法是np.__version__（前后各两个下划线），你代码中的np.<strong>version</strong>是错误格式，修正后代码如下：

   import numpy as np
   print(np.__version__)
   

2. 模块导入冲突：你的dir(np)输出仅显示基础模块属性，无NumPy常用函数/属性，说明当前项目目录下可能存在名为numpy.py的本地文件，导致Python导入了该文件而非官方NumPy库。解决步骤：
   • 删除或重命名当前目录下的numpy.py文件
   • 删除对应的numpy.pyc文件（若存在）
   • 重新运行代码，此时会导入官方NumPy库，dir(np)会显示正常属性列表，np.__version__也能正常输出。

内容的提问来源于stack exchange，提问作者Moaz Azzam