Make sure all blocks are str

master
Dan Howe 3 years ago
parent c2d1e52fe6
commit f21a6b31c3

@ -472,6 +472,7 @@ def process(beach_name, beach_scenario, n_runs, start_year, end_year,
# Create output dataframe # Create output dataframe
df_out = df_in[['beach', 'block', 'profile']] df_out = df_in[['beach', 'block', 'profile']]
df_out['block'] = df_out['block'].astype(str)
# Add info on non-erodable sections # Add info on non-erodable sections
df_out = df_out.assign(min_chainage=np.nan) df_out = df_out.assign(min_chainage=np.nan)
@ -486,7 +487,7 @@ def process(beach_name, beach_scenario, n_runs, start_year, end_year,
df_out = df_out.assign(segment_gaps=False) df_out = df_out.assign(segment_gaps=False)
for b, p, in zip(segment_gaps['block'], for b, p, in zip(segment_gaps['block'],
segment_gaps['profile']): segment_gaps['profile']):
idx = (str(df_out['block']) == str(b)) & (df_out['profile'] idx = (df_out['block'] == str(b)) & (df_out['profile']
== p) == p)
df_out.loc[idx, 'segment_gaps'] = True df_out.loc[idx, 'segment_gaps'] = True

Loading…
Cancel
Save