wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-18 12:20:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: Use context manager to suppress nan-categorical warning

Browse Source
This commit is contained in:
Ana Ruelas
2017-06-02 16:48:26 -04:00
parent 69d1269fc4
commit 6b19e45db4
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/utils/pandas_utils.py
+4 -3
View File
@@ -258,10 +258,11 @@ def categorical_df_concat(df_list, inplace=False):
new_categories = sorted(
set().union(
*(frame[col].cat.categories for frame in df_list)
) - {None}
)
)
for df in df_list:
df[col].cat.set_categories(new_categories, inplace=True)
with ignore_pandas_nan_categorical_warning():
for df in df_list:
df[col].cat.set_categories(new_categories, inplace=True)
return pd.concat(df_list)