Выберите систему координат наложения сетки с помощью matplotlib

Question

Как узнать, какая система координат используется на графике, и как указать, какую систему координат я хочу использовать для наложенной сетки?

from astropy.wcs import WCS
from astropy.io import fits
from astropy.utils.data import get_pkg_data_filename
import matplotlib.pyplot as plt

image_file = get_pkg_data_filename('tutorials/FITS-images/HorseHead.fits')
hdu = fits.open(image_file)[0]
wcs = WCS(hdu.header)

plt.subplot(projection=wcs) 
plt.imshow(hdu.data, origin='lower') 
plt.grid(color='white', ls='solid')
plt.show()

Как мне указать, что Iхотите, чтобы галактические (или экваториальные) координаты использовались для plt.grid?

Answer 1

Вы можете сделать:

from astropy.wcs import WCS
from astropy.io import fits
from astropy.utils.data import get_pkg_data_filename
import matplotlib.pyplot as plt

image_file = get_pkg_data_filename('tutorials/FITS-images/HorseHead.fits')
hdu = fits.open(image_file)[0]
wcs = WCS(hdu.header)

fig = plt.figure(figsize=(12,12))
plt.subplot(projection=wcs) 
plt.imshow(hdu.data, origin='lower') 
ax = fig.gca()
overlay = ax.get_coords_overlay('galactic')
overlay.grid(color='red', ls='dotted')
plt.show()

Как это делается здесь .(Не проверял математику за этим.)