使用astropy软件包修改python脚本

时间:2018-07-14 22:35:32

标签: python astropy

我有一个python脚本,可以拍摄适合的图像并绘制通量与像素的关系。 它从选定的中心点绘制了6条线,它们对应于0、30、60、90、120和150度。 然后,它计算沿每条线的通量。 我想使用2张图像并覆盖它们,然后分别绘制两个图像的每条线以进行比较。 即在第二张图片的0度上绘制的第一张图片的0度。 该脚本有一个用于天文学的astropy软件包,可以在线下载,请尝试anaconda。

脚本如下:

#CODE CREATED BY A.AVISON 20-APR-2018#

import numpy as np
import scipy.ndimage
import matplotlib.pyplot as plt
from astropy.io import fits

#--- FITSNAME ---#
fitsName="fix_N7027.JVLA+eMer.eq.image.fits"

#---get fits image data---#
hdulist = fits.open(fitsName)

imData=hdulist[0].data
imData[np.isnan(imData)] = -0.0005 #== set NaNs to something which is a real number
roughSigma=np.median(imData[np.where(imData>0.0)])#=== takes the median non-zero values as a pseudoRMS
levs=np.arange(5.0,55.0,10.0)*roughSigma

#=== DEFINE CeNTRAL POSTION
y_peak=1043  #in pixels
x_peak=990   #in pixels
cutLen=500.0 #in pixels
#-- Plot...
fig, axes = plt.subplots(ncols=2, figsize=[12,6])
axes[0].imshow(imData,cmap='bone',origin='lower')#'gist_ncar'
axes[0].contour(imData,colors='w',linewidths=0.5,levels=levs)

#-- Extract data values along the line...
# Make a line with "num" points...
numCuts=6
angPerCuts=180.0/float(numCuts)

for ang in range(numCuts):
    x1=x_peak-float(np.floor(cutLen*np.cos(np.radians(float(ang)*angPerCuts))))
    y1=y_peak-float(np.floor(cutLen*np.sin(np.radians(float(ang)*angPerCuts))))

    x0=x_peak+float(np.floor(cutLen*np.cos(np.radians(float(ang)*angPerCuts))))
    y0=y_peak+float(np.floor(cutLen*np.sin(np.radians(float(ang)*angPerCuts))))

    num = int(cutLen)
    x, y = np.linspace(x0, x1, num), np.linspace(y0, y1, num)
    # Extract the values along the line, using cubic interpolation
    zi = 0
    zi = scipy.ndimage.map_coordinates(imData, np.vstack((x,y)))

    axes[0].plot(y,x, '-',linewidth=1.0,alpha=0.8,label=str(float(ang)*angPerCuts)+"degrees")
    axes[0].plot(y1,x1, 'g.',linewidth=1.0,alpha=0.05)
    axes[1].plot(zi,'-',alpha=0.8,linewidth=1.0,label=str(float(ang)*angPerCuts)+"degrees")

axes[1].set_xlim([0,cutLen])

#=== OVERLAY 1/r^x models ===#
axes[1].legend()
plt.show()
hdulist.close()

图已附上。

一个人建议将代码加倍(再次重复)并修改轴!!!我该怎么办?。

0 个答案:

没有答案