fix(mf5to6.py::segment_data_to_period_data): use nseg column instead …

…of dataframe index for segment numbers
aleaf · Sep 21, 2023 · a5be8ee · a5be8ee
1 parent 118615d
commit a5be8ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sfrmaker/mf5to6.py b/sfrmaker/mf5to6.py
@@ -412,7 +412,7 @@ def segment_data_to_period_data(segment_data, reach_data):
     # get rno for reach 1s
     segroups = rd.groupby('iseg')
     reach1IDs = segroups.first()['rno'].to_dict()
-    prd['rno'] = [reach1IDs[i + 1] for i in prd.index]
+    prd['rno'] = [reach1IDs[seg] for seg in prd['nseg']]
     # prd['rainfall'] = np.random.randn(len(prd))
 
     #  distribute other variables to all reaches