Skip to content

Commit bf02be0

Browse files
authored
fix(flopy/utils/sfroutputfile.py::SfrFile.get_results): refactor deprecated DataFrame.append() call to pandas.concat() (#2491)
Refactor deprecated DataFrame.append() call to pandas.concat()
1 parent eb91c4d commit bf02be0

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

autotest/test_sfr.py

+8
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import matplotlib
88
import matplotlib.pyplot as plt
99
import numpy as np
10+
import pandas as pd
1011
import pytest
1112
from modflow_devtools.markers import requires_exe, requires_pkg
1213

@@ -675,6 +676,13 @@ def test_SfrFile(function_tmpdir, sfr_examples_path, mf2005_model_path):
675676
assert df.column.values[0] == 169
676677
assert df.Cond.values[0] == 74510.0
677678
assert df.gw_head.values[3] == 1.288e03
679+
680+
segments = [1, 2, 3]
681+
reaches = [3, 3, 3]
682+
subset_df = sfrout.get_results(segments, reaches)
683+
expected = pd.concat([df.loc[(df['segment'] == s) & (df['reach'] == r)]
684+
for s, r in zip(segments, reaches)])
685+
subset_df.equals(expected)
678686

679687
sfrout = SfrFile(sfr_examples_path / "test1tr.flw")
680688
assert sfrout.ncol == 16, sfrout.ncol

flopy/utils/sfroutputfile.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def get_results(self, segment, reach):
245245
for s, r in locsr:
246246
srresults = self._get_result(s, r)
247247
if len(srresults) > 0:
248-
results = results.append(srresults)
248+
results = pd.concat([results, srresults])
249249
else:
250250
print(f"No results for segment {s}, reach {r}!")
251251
return results

0 commit comments

Comments
 (0)