我有一个数组A,它的形状是(N,N,K),我想计算另一个数组B形状相同,其中B[:,:,i]=np.linalg.inv(A[:,:,i])。作为解决方案,我看到了map和for循环,但我想知道numpy是否提供了执行此操作的功能(我试过np.apply_over_axes但它似乎只能处理一维数组)。使用for循环:B=np.zeros(shape=A.shape)foriinrange(A.shape[2]):B[:,:,i]=np.linalg.inv(A[:,:,i])使用map:B=np.asarray(map(np.linalg.inv,np.squeeze(np.dspli
给定一个3D数组a,我想对其第一个索引中的每个元素调用np.tranpose。例如,给定数组:array([[[1,1,1,1],[1,1,1,1],[1,1,1,1]],[[2,2,2,2],[2,2,2,2],[2,2,2,2]],[[3,3,3,3],[3,3,3,3],[3,3,3,3]])我要:array([[[1,1,1],[1,1,1],[1,1,1],[1,1,1]],[[2,2,2],[2,2,2],[2,2,2],[2,2,2]],[[3,3,3],[3,3,3],[3,3,3],[3,3,3]]])本质上,我想转置数组中的每个元素。我试图reshape它,但找不到