FloPy

Quick demo on how to use FloPy to save array data as a binary file



In [1]:

    
%matplotlib inline
import os 
import sys
import shutil
import flopy
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

print(sys.version)
print('numpy version: {}'.format(np.__version__))
print('matplotlib version: {}'.format(mpl.__version__))
print('flopy version: {}'.format(flopy.__version__))









    



3.6.0 |Anaconda 4.3.0 (x86_64)| (default, Dec 23 2016, 13:19:00) 
[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)]
numpy version: 1.11.3
matplotlib version: 2.0.0
flopy version: 3.2.6



In [2]:

    
nlay,nrow,ncol = 1,20,10
model_ws = os.path.join("data","binary_data")

if os.path.exists(model_ws):
    shutil.rmtree(model_ws)
    
precision = 'single' # or 'double'
dtype = np.float32 # or np.float64

mf = flopy.modflow.Modflow(model_ws=model_ws)
dis = flopy.modflow.ModflowDis(mf, nlay=nlay, nrow=nrow, ncol=ncol, delr=20, delc=10)

Create a linear data array



In [3]:

    
# create the first row of data
b = np.linspace(10, 1, num=ncol, dtype=dtype).reshape(1, ncol)

# extend data to every row
b = np.repeat(b, nrow, axis=0)

# print the shape and type of the data
print(b.shape)

Plot the data array



In [4]:

    
mm = flopy.plot.ModelMap(model=mf)
v = mm.plot_array(b)
mm.plot_grid()
plt.colorbar(v, shrink=0.5)









    Out[4]:





<matplotlib.colorbar.Colorbar at 0x10f225e10>

Write the linear data array to a binary file



In [5]:

    
text = 'head'

# write a binary data file
pertim = dtype(1.0)
header = flopy.utils.BinaryHeader.create(bintype=text, precision=precision,
                                         text=text, nrow=nrow, ncol=ncol,
                                         ilay=1, pertim=pertim,
                                         totim=pertim, kstp=1, kper=1)
pth = os.path.join(model_ws, 'bottom.bin')
flopy.utils.Util2d.write_bin(b.shape, pth, b, header_data=header)

Read the binary data file



In [6]:

    
bo = flopy.utils.HeadFile(pth, precision=precision)
br = bo.get_data(idx=0)

Plot the data that was read from the binary file



In [7]:

    
mm = flopy.plot.ModelMap(model=mf)
v = mm.plot_array(br)
mm.plot_grid()
plt.colorbar(v, shrink=0.5)









    Out[7]:





<matplotlib.colorbar.Colorbar at 0x10f3e0b00>

Plot the difference in the two values



In [8]:

    
mm = flopy.plot.ModelMap(model=mf)
v = mm.plot_array(b-br)
mm.plot_grid()
plt.colorbar(v, shrink=0.5)









    Out[8]:





<matplotlib.colorbar.Colorbar at 0x10f538cf8>



In [ ]: