numpy.astype#

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

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

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

参数:
xndarray

用于转换的输入 NumPy 数组。此处明确不支持 array_likes（类数组对象）。

dtypedtype

结果的数据类型。

copybool, optional

指定当所选的 dtype 与输入数组 x 的数据类型匹配时，是否复制数组。如果为 True，则必须始终返回一个新分配的数组。如果为 False 且指定的 dtype 与输入数组的数据类型匹配，则必须返回输入数组本身；否则，必须返回一个新分配的数组。默认为 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
在此页面