Python Numpy Create Diagonal Matrix From Commando Game 2