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。但由于矩阵无论如何都要独立处理，这应该不是什么大问题。