結果 : find diagonal of matrix python