o sf@sddlZddZdS)NcCsV|jd}t|D]}t|d|D]}d|||f<qq t|}t|t||S)Nr)shaperangenpsumprint)matrixnij total_sumr 4/workspaces/Masterarbeit/14_diag_zero/output_5_0S.py diag_zeros   r)numpyrrr r r rs