python code for eigenvalues and eigenvectors with numpy