如何对一个数组进行函数计算输出另一个数组

在Python中，你可以使用numpy库来进行数组运算并生成新的数组。假设有一个名为`arr`的数组，你想对每个元素应用一个函数（如平方、立方或其他自定义函数），可以按照以下步骤操作：

1. 首先，你需要导入numpy模块：

   import numpy as np

2. 然后定义你要使用的函数，比如平方函数：

   def square(x):
       return x**2

3. 对数组`arr`中的每个元素应用这个函数，使用`numpy.apply_along_axis()`或`numpy.vectorize()`：
- `apply_along_axis()`适用于沿特定轴操作数组：

     new_arr = np.apply_along_axis(square, axis=0, arr=arr)

- `vectorize()`将普通函数转换为可用于整个数组的矢量化版本：

     square_vec = np.vectorize(square)
     new_arr = square_vec(arr)

这两个方法都会创建一个新的数组，新数组的每个元素都是原数组对应元素经过函数处理的结果。