From 1cec7239cd2fff6d590a09378ccd3b3a454140dd Mon Sep 17 00:00:00 2001 From: Mikko Kotila Date: Sat, 20 Apr 2024 16:54:15 +0300 Subject: [PATCH] fix .drop() axis --- wrangle/df/df_corr_randomforest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wrangle/df/df_corr_randomforest.py b/wrangle/df/df_corr_randomforest.py index 0d841b9..d9f2577 100644 --- a/wrangle/df/df_corr_randomforest.py +++ b/wrangle/df/df_corr_randomforest.py @@ -21,7 +21,7 @@ def df_corr_randomforest(data, y, destructive=True): if destructive is False: data = data.copy(deep=True) - x = data.drop(y, 1).values + x = data.drop(y, axis=1).values labels = data.drop(y, axis=1).columns y = data[y].values