본문 바로가기

python

python ndarray flip rotation

fits image invert & two fits images plot



numpy.fliplr, flipud


http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.fliplr.html





# making fits image inverted in x-y axes
# python invertxyfits.py inimage outimage
import astropy.io.fits as fits
import sys
import matplotlib.pyplot as plt
import aplpy
import numpy as np
import os


im=sys.argv[1]
outim=sys.argv[2]

ad=fits.getdata(im)
ah=fits.getheader(im)

os.system('rm '+outim)
y=ad.shape[0]
x=ad.shape[1]

#left-right flip, 
td=np.fliplr(ad)

#up-donw flip
#td=np.flipud(td)


fits.writeto(outim,td,header=ah)

print 'XY invert image from', im, 'to', outim,'is created'

outd=fits.getdata(outim)

plt.clf()
fig=plt.figure(figsize={20,12})
ax=fig.add_subplot(121)

ax.imshow(ad)
ax.set_title('before '+im)

ax=fig.add_subplot(122)
ax.imshow(outd)#,cmap=plt.cm.gray)
ax.set_title('after '+outim)

fig.show()




'python' 카테고리의 다른 글

anaconda virtualenv setting using 'conda'  (0) 2015.06.15
.pythonrc : python startup  (0) 2015.05.28
secat2reg.py  (0) 2015.05.12