numpy.astype#

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

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

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

参数:
xndarray

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

dtypedtype

结果的数据类型。

copy布尔值, 可选

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

device字符串, 可选

放置返回数组的设备。默认值：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
在本页