df_cols = ["doc", "context", "position", "token"] + meta_cols
dfs.append(pd.DataFrame(OrderedDict(zip(df_cols, df_windata))))
return pd.concat(dfs).set_index(["doc", "context", "position"]).sort_index()
elif not with_metadata:
return {dl: [win["token"] for win in windows]
for dl, windows in kwic.items()}
else:
After Change
df_cols = ["doc", "context", "position", "token"] + meta_cols
dfs.append(dt.Frame(OrderedDict(zip(df_cols, df_windata))))
kwic_df = dt.rbind(*dfs)
return kwic_df[:, :, dt.sort("doc", "context", "position")]
elif not with_metadata:
return {dl: [win["token"] for win in windows]
for dl, windows in kwic.items()}
else:
