numpy.astype#

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

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

此函数是 numpy.ndarray.astype 的 Array API 兼容替代方法。

参数:
xndarray

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

dtypedtype

结果的数据类型。

copybool, optional

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

devicestr，可选

用于放置返回数组的设备。默认值：None。仅用于 Array-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
在此页面