Cómo crear una matriz de identidad con numpy

Question

¿Cómo puedo crear una identidad matriz con numpy? ¿Existe una sintaxis más simple que

numpy.matrix(numpy.identity(n)) 

Answer 1

Aquí hay una sintaxis más simple:

np.matlib.identity(n) 

Y aquí es una sintaxis aún más simple que corre mucho más rápido:

In [1]: n = 1000 
In [2]: timeit np.matlib.identity(n) 
100 loops, best of 3: 8.78 ms per loop 
In [3]: timeit np.matlib.eye(n) 
1000 loops, best of 3: 695 us per loop 

Answer 2

no creo que hay una solución más simple . Puede hacerlo ligeramente más eficiente , sin embargo:

numpy.matrix(numpy.identity(n), copy=False) 

Esto evita innecesariamente la copia de los datos.

Answer 3

también np.eye se puede utilizar para crear una matriz de identidad (A).

Por ejemplo,

>>> np.eye(2, dtype=int) 
array([[1, 0], 
     [0, 1]]) 
>>> np.eye(3, k=1) 
array([[ 0., 1., 0.], 
     [ 0., 0., 1.], 
     [ 0., 0., 0.]])