我有一个数据框,例如:
我需要用一个使用sklearn.LinearRegression值的线性模型来填充来自年龄的NA值。
它可以在新的列中,比如"Age_LM“或者在同一列中。
我已经在列表中有来自线性回归模型的值。我只想知道在Age列的每个NA上替换这些值的最佳方法。
发布于 2019-07-02 04:52:24
找到了这个解决方案:
X['Age_Pred'] = linear_model.predict(X.drop('Age', axis = 1))
X['Age_LM'] = np.where(X['Age']>0, X['Age'], X['Age_Pred'])
https://stackoverflow.com/questions/56845304
复制相似问题