numpy.astype#

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

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

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

参数：:
xndarray

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

dtypedtype

结果的数据类型。

copybool, 可选

指定当指定的数据类型与输入数组 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
本页内容