How to diagonally populate a matrix using numpy diagonal