Skip to content

Commit b9eeb7b

Browse files
committed
exlcude boolean type columns from being overwritten with None/nan
1 parent ef938cf commit b9eeb7b

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

flasc/data_processing/filtering.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,9 +313,14 @@ def filter_by_condition(
313313
if not apply_filters_to_df:
314314
df_in = df_in.copy()
315315

316-
# Mark data as faulty on the dataframe
316+
# Mark data as faulty on the dataframe, omitting boolean type columns.
317317
N_pre = [df_get_no_faulty_measurements(df_in, tii) for tii in ti]
318-
df_out = df_mark_turbdata_as_faulty(df=df_in, cond=condition, turbine_list=ti)
318+
df_out = df_mark_turbdata_as_faulty(
319+
df=df_in,
320+
cond=condition,
321+
turbine_list=ti,
322+
exclude_columns=df_in.select_dtypes(include=["bool"]).columns.tolist(),
323+
)
319324

320325
# Print the reduction in useful data to the console, if verbose
321326
if verbose:
@@ -1381,6 +1386,7 @@ def filter_df_by_faulty_impacting_turbines(df, ti, df_impacting_turbines, verbos
13811386
df=df,
13821387
cond=test_turbine_impacted,
13831388
turbine_list=[ti],
1389+
exclude_columns=df.select_dtypes(include=["bool"]).columns.tolist(),
13841390
)
13851391
N_post = df_get_no_faulty_measurements(df_out, ti)
13861392

0 commit comments

Comments
 (0)