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()