def add_fits_units(filein,bunit): hdu = fits.open(filein) header = hdu[0].header header.set('BUNIT', 'K') hdu.writeto(filein,clobber=True) hdu.close()