Python程序设计实验 线性代数操作

【问题描述】
根据以下要求，代码实现线性代数操作。
a = np.array([[1.,2.],[3.,4.]])
y = np.array([[5.],[7.]])
(1) 输出创建的数组a
(2) 输出数组a的转置
(3) 输出形状为(2,2)的对角矩阵b
(4) 输出对角矩阵的迹
(5) 求解数组a和数组y的解
【输入形式】
【输出形式】输出完每一题答案后需换行输下一题答案

import numpy as np
def main():
    a = np.array([[1., 2.], [3., 4.]])
    y = np.array([[5.], [7.]])
    print(a)
    print(a.T)
    b=np.eye(2)
    print(b)
    print(b.trace())
    print(np.linalg.solve(a,y))
if __name__ == '__main__':
    main()
 

 输出为：

[[1. 2.]
 [3. 4.]]
[[1. 3.]
 [2. 4.]]
[[1. 0.]
 [0. 1.]]
2.0
[[-3.]
 [ 4.]]