具有numpy的3D数组的指数

Question

给出一个包含N个MxM正方形阵列的NxMxM阵列，我该如何计算MxM阵列的N个指数。

from scipy.linalg import expm
U = np.random.rand(10,6,6) #Given 10 six by six arrays calculate the following:
exp = expm(U)
>>> [expm(U[0]),expm(U[1])...expm(U[9])]

我想避免for循环并使用numpy操作。

Answer 1

您可以使用列表理解：

exp = [expm(x) for x in U]

由于expm仅接受单个方矩阵，因此矩阵循环将运行Python。但是由于矩阵无论如何都要独立处理，所以这不是什么大问题。