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?

[wpbonelli]

yeah, disv/u weren't covered. sorry @aestrad-intera we are discovering the state of all this as we go. does #2566 solve it for you?

[wpbonelli]

@aestrad-intera please reopen if this does not solve the issue for you

[aestrad-intera]

It did not.

…

On Mon, Aug 4, 2025, 8:59 AM wpbonelli ***@***.***> wrote: *wpbonelli* left a comment (modflowpy/flopy#2531) <#2531 (comment)> yeah, disv/u weren't covered. sorry @aestrad-intera <https://github.com/aestrad-intera> we are discovering the state of all this as we go. does #2566 <#2566> solve it for you? — Reply to this email directly, view it on GitHub <#2531 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/A52MYOPVPC76MUV7OUSDDIT3L5RL5AVCNFSM6AAAAAB7I4J4V6VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZTCNJQHAZTOMRYG4> . You are receiving this because you were mentioned.Message ID: ***@***.***>