Python Numpy Create Diagonal Matrix From Commando