Python numpy ndarray.astype用法及代碼示例

本文簡要介紹 python 語言中 numpy.ndarray.astype 的用法。

用法:

ndarray.astype(dtype, order='K', casting='unsafe', subok=True, copy=True)

數組的副本，強製轉換為指定類型。

參數：

dtype： str 或 dtype

數組轉換為的類型代碼或數據類型。

order： {‘C’、‘F’、‘A’、‘K’}，可選

控製結果的內存布局順序。 “C”表示 C 順序，“F”表示 Fortran 順序，如果所有數組都是 Fortran 連續的，則“A”表示“F”順序，否則為“C”順序，“K”表示與數組元素出現的順序接近盡可能在內存中。默認為“K”。

casting： {‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}，可選

控製可能發生的數據類型轉換。默認為 ‘unsafe’ 以實現向後兼容性。

• ‘no’ 表示根本不應該轉換數據類型。

• ‘equiv’ 表示隻允許更改字節順序。

• ‘safe’ 意味著隻允許可以保留值的強製轉換。

• ‘same_kind’ 表示隻允許安全類型轉換或類型中的類型轉換，如 float64 到 float32。

• ‘unsafe’ 表示可以進行任何數據轉換。

subok： 布爾型，可選

如果為 True，則子類將為 passed-through(默認)，否則返回的數組將被強製為 base-class 數組。

copy： 布爾型，可選

默認情況下， astype 總是返回一個新分配的數組。如果設置為 false，並且numpy.dtype,次序， 和subok滿足要求，則返回輸入數組而不是副本。

返回：

arr_t： ndarray

除非numpy.copy為 False 並且滿足返回輸入數組的其他條件(參見說明numpy.copy輸入參數)，arr_t是與輸入數組具有相同形狀的新數組，具有 dtype，順序為numpy.dtype,次序.

拋出：

ComplexWarning

從 complex 轉換為 float 或 int 時。為避免這種情況，應使用 a.real.astype(t) 。

注意：

例子：

>>> x = np.array([1, 2, 2.5])
>>> x
array([1. ,  2. ,  2.5])

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