Cómo convertir la matriz numpy flotante 2D a 2D int numpy array?

Question

Cómo convertir una matriz numpy real a una matriz int numpy? Intenté usar el mapa directamente en la matriz pero no funcionó.

Answer 1

Utilice el método astype.

>>> x = np.array([[1.0, 2.3], [1.3, 2.9]]) 
>>> x 
array([[ 1. , 2.3], 
     [ 1.3, 2.9]]) 
>>> x.astype(int) 
array([[1, 2], 
     [1, 2]]) 

Answer 2

Algunas funciones numpy de cómo controlar el redondeo: rint, floor, trunc, ceil. dependiendo de cómo desee rodear los flotadores, arriba, abajo o al int más cercano.

>>> x = np.array([[1.0,2.3],[1.3,2.9]]) 
>>> x 
array([[ 1. , 2.3], 
     [ 1.3, 2.9]]) 
>>> y = np.trunc(x) 
>>> y 
array([[ 1., 2.], 
     [ 1., 2.]]) 
>>> z = np.ceil(x) 
>>> z 
array([[ 1., 3.], 
     [ 2., 3.]]) 
>>> t = np.floor(x) 
>>> t 
array([[ 1., 2.], 
     [ 1., 2.]]) 
>>> a = np.rint(x) 
>>> a 
array([[ 1., 2.], 
     [ 1., 3.]]) 

Para hacer uno de esto en a int, o uno de los otros tipos de numpy, astype (como respondidas por BrenBern):

a.astype(int) 
array([[1, 2], 
     [1, 3]]) 

>>> y.astype(int) 
array([[1, 2], 
     [1, 2]]) 

Answer 3

puede utilizar np.int_:

>>> x = np.array([[1.0, 2.3], [1.3, 2.9]]) 
>>> x 
array([[ 1. , 2.3], 
     [ 1.3, 2.9]]) 
>>> np.int_(x) 
array([[1, 2], 
     [1, 2]]) 

Answer 4

Si no está seguro de que su entrada va a ser una matriz Numpy, puede usar asarray con dtype=int en lugar de astype:

>>> np.asarray([1,2,3,4], dtype=int) 
array([1, 2, 3, 4]) 

Si la matriz de entrada ya tiene el dtype correcta, asarray evita la copia matriz mientras astype no (a menos que se especifique copy=False):

>>> a = np.array([1,2,3,4]) 
>>> a is np.asarray(a) # no copy :) 
True 
>>> a is a.astype(int) # copy :(
False 
>>> a is a.astype(int, copy=False) # no copy :) 
True