Come convertire l'array numpy float 2D in array int numpy 2D?

Question

Come convertire la matrice numpy reale in array int numpy? Ho provato ad usare la mappa direttamente sull'array ma non ha funzionato.

Answer 1

Utilizzare il metodo 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

Alcune funzioni NumPy per come controllare l'arrotondamento: rint, floor, trunc, ceil. a seconda di come si desidera arrotondare i galleggianti, su, giù o verso l'int più vicino.

>>> 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.]]) 

Per fare uno di questo a int, o uno degli altri tipi di NumPy, astype (come risposta da BrenBern):

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

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

Answer 3

è possibile utilizzare 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

Se non si è sicuri che l'input sarà un array Numpy, è possibile utilizzare asarray con dtype=int anziché astype:

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

Se l'array di input ha già la DTYPE corretta, asarray evita la copia matrice mentre astype non (se non si specifica 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