import pylab as plt
from mw_plot import MWPlot, MWSkyMap
from astropy import units as  u

# setup a mw-plot instance of bird's eyes view of the disc
mw1 = MWPlot(radius=20 * u.kpc, center=(0, 0)*u.kpc, unit=u.kpc, coord='galactocentric', rot90=2, grayscale=False, annotation=False)
mw2 = MWPlot(radius=20 * u.kpc, center=(0, 0)*u.kpc, unit=u.kpc, coord='galactocentric', rot90=2, grayscale=True, annotation=False)
mw3 = MWSkyMap()

# setup subplots with matplotlib
fig = plt.figure(figsize=(15, 15))
ax1 = fig.add_subplot(221)
ax2 = fig.add_subplot(222)
ax3 = fig.add_subplot(212)

# transform the subplots with different style
mw1.transform(ax1)
mw2.transform(ax2)
mw3.transform(ax3)

plt.tight_layout()