在 NumPy 中,使用 astype 函数可以将数组的数据类型转换为指定的类型。具体地说,将 np.uint8 类型的数组转换为 np.float32 类型的数组,可以使用以下代码:


import numpy as np

uint8_array = np.array([0, 128, 255], dtype=np.uint8)
float32_array = uint8_array.astype(np.float32) / 255.0

print(float32_array)

输出结果为:


[0.         0.5019608  1.        ]

在这里,我们首先创建一个 np.uint8 类型的数组 uint8_array,包含了三个元素:0、128 和 255。然后,我们使用 astype 函数将其转换为 np.float32 类型的数组 float32_array。为了将数据归一化到 [0, 1] 的范围内,我们还需要将 float32_array 中的每个元素除以 255。


universe_king
3.4k 声望680 粉丝