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

# 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.3))
ax.plot_wireframe(X, Y, data['pres'], rstride=4, cstride=4)

# Show plot on screen
plt.show()