astropy.wcs
astropy.wcs.utils.
add_stokes_axis_to_wcs
Add a new Stokes axis that is uncorrelated with any other axes.
WCS
The WCS to add to
Index of the WCS to insert the new Stokes axis in front of. To add at the end, do add_before_ind = wcs.wcs.naxis The beginning is at position 0.