三维numpy数组中的列的范数

Question

给定形状为P的numpy数组(n, m, m)，我希望得到形状为Q的{​​{1}} n (n, m) Q[i, j] = np.linalg.norm(P[i][:,j])。我尝试使用Q = np.fromfunction(lambda i, j: np.linalg.norm(P[i][:,j]), (n, m), dtype=float)，但我收到了错误

TypeError: only integer arrays with one element can be converted to an index

我有一个很大的阵列，所以我寻找的东西没有明显的时间损失。