如何生成对角矩阵 numpy.diag