Numpy - 计算对角线的乘积

Question

新手问题：假设我有一个矩阵A，是否有一个numpy函数来计算A的对角元素的乘积？

EG：

其中，X（i，i）是矩阵的对角乘积。

Answer 1

您可以使用diagonal和prod方法：

import numpy as np
a = np.matrix([[1, 1, 1],
             [1, 2, 3],
             [3, 3, 3]])

prod_diag = a.diagonal().prod()
print(prod_diag)
# gives 6 as answer

diagonal将矩阵的对角线分量作为一维数组返回，prod计算数组所有元素的乘积。