How to convert 2D float numpy array to 2D int numpy array?

前端 未结 4 939
情话喂你
情话喂你 2020-12-22 18:08

How to convert real numpy array to int numpy array? Tried using map directly to array but it did not work.

4条回答
  •  执笔经年
    2020-12-22 18:43

    Some numpy functions for how to control the rounding: rint, floor,trunc, ceil. depending how u wish to round the floats, up, down, or to the nearest int.

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

    To make one of this in to int, or one of the other types in numpy, astype (as answered by BrenBern):

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

提交回复
热议问题