wcs 0.2.3

Implementation of the FITS World Coordinate System (WCS)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

from astropy.io import fits
from astropy.wcs import WCS
from astropy.coordinates import SkyCoord
import astropy.units as u
import glob
import numpy as np

input_files = glob.glob('./examples/*.fits')
print(input_files)

for filename in input_files:
    # Load the FITS hdulist using astropy.io.fits
    hdulist = fits.open(filename)

    # Parse the WCS keywords in the primary HDU
    w = WCS(hdulist[0].header)

    naxis1 = hdulist[0].header.get("NAXIS1")
    naxis2 = hdulist[0].header.get("NAXIS2")
    
    # Get random sky coordinates
    X = np.random.random(100) * naxis1
    Y = np.random.random(100) * naxis2

    coord = w.wcs_pix2world(np.vstack((X, Y)).T, 0)
    ra, dec = ((coord[:, 0] * u.deg).to_value(u.rad), (coord[:, 1] * u.deg).to_value(u.rad))

    tab = np.column_stack((ra, dec, X, Y))
    np.savetxt(filename + ".csv", tab, delimiter=",")