mapping back any sklearn result to the original dataframe

Question

I'd like to analyze the predicted values of my random forest results in excel with the original test data as a reference.

The predicted result comes in an array as i use this:

predict = rf.predict(test[columns])

how do I map back the predicted results to the original dataset using pandas?

Regards, galeej

I think you need to explain your scenario a little more clearly. Maybe put in some example data as well. — Little Bobby Tables, Dec 19 '16 at 09:17
You should be able to just add this back as a new column: `test['prediction'] = rf.predict(test[columns])` — EdChum, Dec 19 '16 at 10:10
My apologies for the delayed reply...@EdChum your method worked... thanks! — galeej, Dec 31 '16 at 06:28

galeej · Accepted Answer · 2017-02-08T06:37:03.463

2

EdChum's answered the question.

(Edit) His answer:

You should be able to just add this back as a new column: test['prediction'] = rf.predict(test[columns])

My apologies for the delay.

Regards, Sunil

edited Feb 08 '17 at 06:37

answered Feb 08 '17 at 06:23

galeej

1 Answers1