提交 969827a7 编写于 作者: J jrzaurin

adapting the code for crossed columns being a list of tuples

上级 3db75fae
......@@ -5,13 +5,15 @@ from sklearn.preprocessing import StandardScaler
from ..wdtypes import *
def prepare_wide(df:pd.DataFrame, wide_cols:List[str], crossed_cols:List[Tuple[str,str]],
def prepare_wide(df:pd.DataFrame, wide_cols:List[str], crossed_cols,
already_dummies:Optional[List[str]]=None) -> np.ndarray:
df_wide = df.copy()[wide_cols]
crossed_columns = []
for cols in crossed_cols:
cols = list(cols)
colname = '_'.join(cols)
df_wide[colname] = df_wide[cols].apply(lambda x: '-'.join(x), axis=1)
crossed_columns.append(colname)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册