重邮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)
    a_t = np.array([[1.,3.],[2.,4.]])
    print(a_t)
    b = np.eye(2)
    print(b)
    tr_b = b[1][1]+b[0][0]
    print(tr_b)
    print(np.linalg.solve(a,y))
if __name__ == "__main__":
    main()