Nrthugu

I would like to assign a commented out Well ID number to my well file, (also do the same for ghb cells) but I cannot find anything on how to do so.

I wrote something to create my own ghb file but if I try to load it back into my flopy mf class, and later write it out my other packages with mf.write_input() the comments do not stay and it gets overwritten.

I know in mf.wrtie_input() I can specify what packages to write out, and if I take away the ghb file I made earlier (or well file) then the original file does not get written over which is good.

But I would like to know if there is a way to straight up add comments to the stress_period_data for each package so I can keep it all contained in the flopy class.

Thanks

Like in this example, you can extend the default dtype to include extra attributes that the MfList instance will carry to the write:

well_dtype = [('k', '<i8'), ('i', '<i8'), ('j', '<i8'),('flux', '<f4'), ('wel_id', object)]

stress_period_data = np.zeros((3), dtype=well_dtype)

wel = flopy.modflow.ModflowWel(m, stress_period_data=stress_period_data, dtype=well_dtype)

I'm not sure of an easy way to load an existing wel package with extra attributes - just FYI

The only way I know to carry remarks over from an existing package is to open the file in read mode, create a pandas DataFrame with the column data, and build a new package out of it. Here is an example:

import os

import pandas as pd

import flopy.modflow as fpm

from collections import OrderedDict





pak_nam = 'drn'

mf_version = 'mfnwt'



# the model from which the DRN package will be copied

inmod = fpm.Modflow.load('10kTDS.nam',

                     model_ws=r'..10kTDS',

                     version=mf_version, 

                     load_only=['drn'], 

                     check=False)



# the model where the new DRN package will be attached

mf = fpm.Modflow.load('ss2010.nam',

                      model_ws=os.path.join('..', 'ss2010'),

                      version=mf_version,

                      load_only=['dis', 'bas6'],

                      check=False)



# read the contents of the DRN package

with open(inmod.drn.fn_path, 'r') as f:

    lines = f.readlines()



# create pandas DataFrame

data = 

for line in lines[3:]:

    pieces = line.strip().split('#')

    t = pieces[0].strip().split()

    remark = pieces[-1]

    if t[0] == '-1':

        break

    else:

        data.append([int(t[0]),

                     int(t[1]),

                     int(t[2]),

                     float(t[3]),

                     float(t[4]),

                     '# ' + remark.strip()])

pak_df = pd.DataFrame(data,

                      columns=['k', 'i', 'j', 'alt_va', 'cond', 'remark'])

pak_df.loc[:, ['k', 'i', 'j']] -= 1



# specify data format

formats = OrderedDict([('k', '{:>10d}'.format),

                       ('i', '{:>10d}'.format),

                       ('j', '{:>10d}'.format),

                       ('alt_va', '{:>.2F}'.format),

                       ('cond', '{:>15.6E}'.format),

                       ('remark', '{>:50}'.format)])



# create new stress period data: for numpy record array use DataFrame.to_records()

pak_spd = {0: pak_df[list(formats.keys())].to_records(index=False)}



# attach DRN package to new model

pak = fpm.ModflowDrn(mf,

                     stress_period_data=pak_spd,

                     ipakcb=53,

                     options=['NOPRINT'],

                     filenames=os.path.join('..', 'ss2010', 'ss2010.{}'.format(pak_nam)),

                     dtype=pak_spd[0].dtype)



pak.write_file(check=False)