bug: cannot export package stress period data to shapefile

[aestrad-intera]

The following lines of code:

sim = flopy.mf6.MFSimulation.load(sim_ws='run_temp_ws')
gwf = sim.get_model()
mg = gwf.modelgrid
mg.set_coord_info(xoff=5677631.08875377, yoff=18360574.17093341, angrot=50)
chd = gwf.chd
chd.stress_period_data.export(os.path.join('..', '..', 'arcpro', 'shp', 'pumptest_chd.shp'), sparse=True)

Produce this error:

896 from ..export.shapefile_utils import recarray2shp
897 from ..utils.geometry import Polygon

--> 899 df = mfl.get_dataframe(squeeze=squeeze)
900 if "kper" in kwargs or df is None:
901 ra = mfl[kper]

TypeError: MFPandasTransientList.get_dataframe() got an unexpected keyword argument 'squeeze'

It appears that the stress_period_data.export utility is using a deprecated squeeze parameter. What other method is available for exporting list package data to shapefiles?

[wpbonelli]

Thanks for reporting @aestrad-intera there is a patch in the develop branch if you'd like to check it on your end

[aestrad-intera]

@wpbonelli I included you modifications in mflist_export() utils.py but now I am getting a pandas error upon running the same code:

Cell In[2], line 2
1 chd = gwf.chd
----> 2 chd.stress_period_data.export(os.path.join('..', '..', 'arcpro', 'shp', 'pumptest_chd.shp'), sparse=True)

File dependencies\flopy\mf6\data\mfdata.py:375, in MFData.export(self, f, **kwargs)
373 return utils.transient2d_export(f, self, **kwargs)
374 elif self.data_type == DataType.transientlist:
--> 375 return utils.mflist_export(f, self, **kwargs)
376 return utils.transient2d_export(f, self, **kwargs)

File dependencies\flopy\export\utils.py:922, in mflist_export(f, mfl, **kwargs)
915 row_index_name = "i" if "i" in df.columns else "cellid_row"
916 col_index_name = "j" if "j" in df.columns else "cellid_column"
917 verts = np.array(
918 [
919 # modelgrid.get_cell_vertices(i, df.j.values[ix])
920 # for ix, i in enumerate(df.i.values)
921 modelgrid.get_cell_vertices(i, df[col_index_name].values[ix])
--> 922 for ix, i in enumerate(df[row_index_name].values)
923 ]
924 )
925 ra = df.to_records(index=False)
926 crs = kwargs.get("crs", None)

File ~\AppData\Local\anaconda3\envs\mesa\lib\site-packages\pandas\core\frame.py:4102, in DataFrame.getitem(self, key)
4100 if self.columns.nlevels > 1:
4101 return self._getitem_multilevel(key)
-> 4102 indexer = self.columns.get_loc(key)
4103 if is_integer(indexer):
4104 indexer = [indexer]

File ~\AppData\Local\anaconda3\envs\mesa\lib\site-packages\pandas\core\indexes\base.py:3812, in Index.get_loc(self, key)
3807 if isinstance(casted_key, slice) or (
3808 isinstance(casted_key, abc.Iterable)
3809 and any(isinstance(x, slice) for x in casted_key)
3810 ):
3811 raise InvalidIndexError(key)
-> 3812 raise KeyError(key) from err
3813 except TypeError:
3814 # If we have a listlike key, _check_indexing_error will raise
3815 # InvalidIndexError. Otherwise we fall through and re-raise
3816 # the TypeError.
3817 self._check_indexing_error(key)

KeyError: 'cellid_row'

[wpbonelli]

@aestrad-intera to clarify how to reproduce this, could you pull the develop branch and have a look at the test I added in #2532? The only difference I see is setting coordinate info on the grid which is inconsequential here.

[aestrad-intera]

@wpbonelli The model that I am using uses DISU package, could it be that this package is not covered by the patch?