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

这里的问题是在做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")

posted @ 2019-09-29 22:27  boobo  阅读(11642)  评论(0编辑  收藏  举报
^