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')
[((374, 384), 'matplotlib.pyplot.show', 'plt.show', ([], {}), '()\n', (382, 384), True, 'import matplotlib.pyplot as plt\n'), ((400, 410), 'matplotlib.pyplot.show', 'plt.show', ([], {}), '()\n', (408, 410), True, 'import matplotlib.pyplot as plt\n'), ((419, 445), 'matplotlib.pyplot.figure', 'plt.figure', ([], {'figsize': '(8, 8)'}), '(figsize=(8, 8))\n', (429, 445), True, 'import matplotlib.pyplot as plt\n'), ((638, 648), 'matplotlib.pyplot.show', 'plt.show', ([], {}), '()\n', (646, 648), True, 'import matplotlib.pyplot as plt\n'), ((187, 225), 'os.path.join', 'os.path.join', (['""".."""', '"""data"""', '"""freyberg"""'], {}), "('..', 'data', 'freyberg')\n", (199, 225), False, 'import os\n')]