apply_along_axis传入多个数组参数的应用举例
时间: 2023-02-26 12:07:11 浏览: 397
axis调用实例
numpy 的 `apply_along_axis` 函数允许你在数组的某个轴上使用自定义函数处理数据。这个函数可以传入多个数组参数,以下是一个应用示例:
假设我们有两个 2x2 的数组:
```
import numpy as np
def my_func(x, y):
return x + y
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
```
我们可以使用 `apply_along_axis` 在第一维(行)上对这两个数组应用 `my_func` 函数:
```
result = np.apply_along_axis(my_func, 0, arr1, arr2)
print(result)
# 输出:
# [[ 6 8]
# [10 12]]
```
这个函数对每一行分别调用 `my_func` 函数,并传入相应的行作为 `x` 和 `y` 参数。最后得到的结果是一个 2x2 的数组,其中每个元素是 `my_func` 函数的结果。
阅读全文