Python Numpy Create Diagonal Matrix From Characteristic Polynomial