当前位置 : 主页 > 编程语言 > python >

python 将数组中取某一值的元素全部替换为其他元素的方法

来源:互联网 收集:自由互联 发布时间:2021-06-25
这里的问题是在做House Price Prediction的时候遇到的,尝试对GarageArea做log转化,但是由于有些房子没有车库,所以GarageArea = 0,再通过log转化变成-inf了,所以我想把所有-inf的数据全部再转

这里的问题是在做House Price Prediction的时候遇到的,尝试对GarageArea做log转化,但是由于有些房子没有车库,所以GarageArea = 0,再通过log转化变成-inf了,所以我想把所有-inf的数据全部再转化为0,如题目所示。

除了循环还可以怎么做呢?看下面的代码

train_test[GarageArea] = np.log(train_test[GarageArea])

train_test[GarageArea] = train_test[GarageArea].replace(float("-inf"), 0)

本来以为replace()函数是str专用的函数,结果尝试发现在数组上也可以用,这里比较特殊的是-inf的写法,不能直接写成 -inf 或者 ‘-inf‘,而要写成 float("-inf")

网友评论