# Takes two FITS files and multiplies them together. import numpy import pyfits import os # Changes to the directory where the script is located abspath = os.path.abspath(__file__) dname = os.path.dirname(abspath) os.chdir(dname) infile1 = 'Virgo_A_p2_h3_poly_2.fits' infile2 = 'Virgo_B_p2_h3_poly_2.fits' outfile = 'VirgoAXB.fits' FitsFile1 = pyfits.open(infile1) image1= FitsFile1[0].data header=FitsFile1[0].header FitsFile2 = pyfits.open(infile2) image2= FitsFile2[0].data #image[image < 0.1] = 0.0 image = image1*image2 FitsFile = pyfits.writeto(outfile,image,header=header)