在numpy数组中的多维上应用函数

Question

我有一个numpy数组od维度（343343250）。这是一个图像时间系列，第一个尺寸是x坐标，第二个尺寸是y坐标，第三个尺寸是日期

我尝试在每个像素时间序列上应用一个函数，如下所示：

def f(d):

   #do stuff with d
   return stuff

result = np.zeros((data.shape[0], data.shape[1]), dtype=cp.float32)
for i in range(data.shape[0]):
    for j in range(data.shape[1]:

        d = data[i,j,:]
        result[i,j] = f(d)

我怎样才能以更简单的方式做同样的事情（没有for循环）

Answer 1

使用numpy.apply_along_axis：

import numpy as np
result = np.apply_along_axis(f, axis=2, arr=data)