[python] 如何將2D浮點numpy數組轉換為2D int numpy數組?



1 Answers

關於如何控制舍入的一些numpy函數: rintfloortruncceil 。 取決於你希望如何繞浮點,向上,向下或到最近的整數。

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

為了使其中一個int或numpy中的其他類型之一, astype (由BrenBern回答):

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

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

如何將真正的numpy數組轉換為int numpy數組? 試圖直接使用地圖到數組,但它沒有奏效。




如果你不確定你的輸入是否是一個Numpy數組,你可以用asarray dtype=int而不是astype來使用astype

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

如果輸入數組已經具有正確的asarrayasarray避免數組副本,而astype不會(除非指定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



Related



Tags

python python   numpy