python - 将矩阵的元素转换为 float

Question

这是我正在处理的 data_1 的输出:

print(data_1)  

[['1.0' '1.585' '1.78']
 ['1.0' '1.908' '1.816']
 ['1.0' '1.7' '1.732']
 ..., 
 ['1.0' '1.046' '4.824']
 ['1.0' '1.138' '4.874']
 ['1.0' '1.016' '5']]

我尝试通过简单地使用以下代码将元素转换为 float :

data_2 = float(data_1)   

但是我得到了以下错误

TypeError: only length-1 arrays can be converted to Python scalars

如何将该矩阵的元素转换为 float ？

Answer 1

看起来你有一个 numpy 数组，float 是一个基本的 python 函数，只接受标量作为参数，即它只能将标量转换为 float 。对于 numpy 数组，有一个内置的 astype 函数可用于类型转换:

data_1.astype(float)
# array([[ 1.   ,  1.585,  1.78 ],
#        [ 1.   ,  1.908,  1.816],
#        [ 1.   ,  1.7  ,  1.732],
#        [ 1.   ,  1.046,  4.824],
#        [ 1.   ,  1.138,  4.874],
#        [ 1.   ,  1.016,  5.   ]])