将函数应用于np.ndarray？

Question

我有一个numpy数组，例如一个（a.shape =（2,20,50,50））。 我想在第二个轴上应用一个函数。

我的功能是仅沿第二个轴的每个元素之间的差异，即

res = a[:,i+1,:,:] - a[:,i,:,:] for i in range(20)

我已经用lambda函数尝试过了，但是输出是一个列表。 我希望得到与a形状相同的结果。 这意味着我想拥有res.shape =（2,20,50,50）。

我真的很感谢有人指导我。

先感谢您。

Answer 1

您不需要应用功能。 只需直接减去它们。

res = a[:,1:,:,:]-a[:,0:-1,:,:]

此外，您不会得到（2,20,50,50）ndarray，而会得到（2,19,50,50）。