def pulsar_select(x0, y0, r): pulsar_data = 'pulsar3.lis' t, x, y, E = read_pulsar(pulsar_data) t_signal = [] x_signal = [] y_signal = [] E_signal = [] for i in range(0, len(x)): dist = math.sqrt((x[i] - x0)**2 + (y[i] - y0)**2) if dist < r: t_signal.append(t[i]) x_signal.append(x[i]) y_signal.append(y[i]) E_signal.append(E[i]) return t_signal, x_signal, y_signal, E_signal
# plot_image.py # # Plot the x,y coordinates of each detected X-ray in the pulsar data file from pylab import * from read_pulsar import * from pulsar_select import * pulsar_data = 'pulsar3.lis' # read_pulsar returns time (t), x, y, and energy (E) as vectors t, x, y, E = read_pulsar(pulsar_data) # Alternately, just plot the selected points #x0 = 26276. #y0 = 27943. #r = 150.0 #t, x, y, E = pulsar_select(x0, y0, r) fig = plt.figure() ax = fig.add_subplot(111) hist = ax.hist2d(x,y,bins=512,cmap=cm.hot_r) ax.set_title('X-ray image data, Pulsar 3: binned 512x512') #cb = colorbar(hist[0]) #cb.set_label('counts') show() fig.savefig('hist')