numpy.astype#

numpy.astype(x, dtype, /, *, copy=True, device=None)[source]#

将数组复制到指定的数据类型。

此函数是与数组 API 兼容的替代方案,替代了 numpy.ndarray.astype

参数:
xndarray

要转换的输入 NumPy 数组。此处明确不支持 array_likes

dtypedtype

结果的数据类型。

copybool, 可选

指定当指定的 dtype 与输入数组 x 的数据类型匹配时是否复制数组。如果为 True,则必须始终返回一个新分配的数组。如果为 False 且指定的 dtype 与输入数组的数据类型匹配,则必须返回输入数组;否则,必须返回一个新分配的数组。默认为 True

devicestr, 可选

放置返回数组的设备。默认值:None。仅用于数组 API 互操作性,因此如果传递,则必须为 "cpu"

版本 2.1.0 中的新功能。

返回:
outndarray

具有指定数据类型的数组。

另请参见

ndarray.astype

示例

>>> import numpy as np
>>> arr = np.array([1, 2, 3]); arr
array([1, 2, 3])
>>> np.astype(arr, np.float64)
array([1., 2., 3.])

非复制案例

>>> arr = np.array([1, 2, 3])
>>> arr_noncpy = np.astype(arr, arr.dtype, copy=False)
>>> np.shares_memory(arr, arr_noncpy)
True