#!/usr/bin/env python
# coding: utf-8

# In[1]:


import re
import netCDF4 as nc
import numpy as np
import matplotlib.pyplot as plt

get_ipython().run_line_magic('matplotlib', 'inline')


# In[2]:


f=nc.Dataset('/data/eolson/MEOPAR/SS36runs/runSOG5x5_NewIC/SOG5x5_1ts_20041019_20041023_ptrc_T.nc')
fkeys=f.variables.keys()
for ik in fkeys:
    match = re.search(r'depth.',ik)
    if match:
        zkey=match.group(0)
z=f.variables[zkey][:]
t=f.variables['time_counter'][:]
tz,zt=np.meshgrid((t-t[0])/3600/24,-z[:])


# In[3]:


print(t[0:5]-t[0])


# In[4]:


for ik in fkeys:
    if np.size(f.variables[ik].shape)==4:
        fig, axs = plt.subplots(1,1,figsize=(10,4))
        mesh=plt.pcolormesh(tz[:3600*24*1,:],zt[:3600*24*1,:],f.variables[ik][:3600*24*1,:,2,2].T)
        plt.colorbar()
        plt.title(ik)


# In[ ]:





# In[ ]: