from __future__ import print_function
import os

import numpy as np
import matplotlib.pyplot as plt

import flopy


fb = flopy.modflow.Modflow.load('freyberg', version='mf2005', model_ws=os.path.join('..', 'data', 'freyberg'), verbose=True)

dis = fb.dis

top = fb.dis.top

fb.dis.top.plot(grid=True, colorbar=True)
fb.dis.botm.plot(grid=True, colorbar=True)

fb.dis.plot()
plt.show()

fb.dis.plot()
plt.show()


fig = plt.figure(figsize=(8, 8))
ax = fig.add_subplot(1,2,1, aspect='equal')
fb.dis.top.plot(grid=True, axes=ax,  colorbar=True)
ax = fig.add_subplot(1,2,2, aspect='equal')
fb.dis.botm.plot(grid=True, axes=ax, colorbar=True)
plt.show()

print('this is the end my friend')
[((20, 0, 20, 10), 'matplotlib.pyplot.show', 'plt.show', ({}, {}), '()', True, 'import matplotlib.pyplot as plt\n'), ((23, 0, 23, 10), 'matplotlib.pyplot.show', 'plt.show', ({}, {}), '()', True, 'import matplotlib.pyplot as plt\n'), ((26, 6, 26, 32), 'matplotlib.pyplot.figure', 'plt.figure', (), '', True, 'import matplotlib.pyplot as plt\n'), ((31, 0, 31, 10), 'matplotlib.pyplot.show', 'plt.show', ({}, {}), '()', True, 'import matplotlib.pyplot as plt\n'), ((10, 71, 10, 109), 'os.path.join', 'os.path.join', ({(10, 84, 10, 88): '""".."""', (10, 90, 10, 96): '"""data"""', (10, 98, 10, 108): '"""freyberg"""'}, {}), "('..', 'data', 'freyberg')", False, 'import os\n')]