import gpplot as gplt
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm

# Set fontsizes
plt.rcParams.update({'axes.titlesize':16,'axes.labelsize':14,'xtick.labelsize':12,'ytick.labelsize':12})

# Import data and extract fields into dictionary
data = gplt.dataFromFile('RayleighTaylor0128_3_002975.h5')

# Create meshgrid from individual axes
X,Y = np.meshgrid(data['x'],data['y'])

# Create figure and 3D axes
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
ax.set_box_aspect((0.25,1,0.1))
surf = ax.plot_surface(X, Y, data['eint'], cmap=cm.CMRmap, linewidth=0, antialiased=False)
fig.colorbar(surf,shrink=0.4)

# Show plot on screen
plt.show()