Python Numpy Create Diagonal Matrix From Tripod