python 求一个3*3矩阵对角线元素的乘积之和
时间: 2023-09-12 17:08:13 浏览: 143
Python人工智能课程 AI算法课程 Python机器学习与深度学习 1.矩阵和线性代数_Python 共90页.pdf
可以使用 NumPy 库来实现:
```python
import numpy as np
# 创建一个3*3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 提取对角线元素
diagonal = np.diagonal(matrix)
# 计算对角线元素的乘积之和
result = np.prod(diagonal)
print(result) # 输出结果为 45
```
首先创建一个 3*3 的矩阵,然后使用 `np.diagonal()` 函数提取对角线元素,再使用 `np.prod()` 函数计算乘积之和。最终的结果为 45。
阅读全文