Нарезка графика

Question

Я создал график в python, но теперь мне нужно взять раздел графика и развернуть его, используя небольшой диапазон исходных данных, но я не знаю, как найти номер строки результаты, которые формируют диапазон или как я могу создать граф, используя только эти результаты, образуют файл. Это код, у меня есть для графа:

import numpy as np 
import matplotlib.pyplot as plt 

#variable for data to plot 
spec_to_plot = "SN2012fr_20121129.42_wifes_BR.dat" 

#tells python where to look for the file 
spec_directory = '/home/fh1u16/Documents/spectra/' 
data = np.loadtxt(spec_directory + spec_to_plot, dtype=np.float) 

x = data[:,0] 
y = data[:,1] 

plt.plot(x, y) 
plt.xlabel("Wavelength") 
plt.ylabel("Flux") 
plt.title(spec_to_plot) 
plt.show() 

редактировать: данные между 3.5e + 3 и 9.9e + 3 в первой колонке, мне нужно использовать только данные между 5.5e + 3 и 6е +3 для построения другого графика, но это относится только к первому столбцу. Надеюсь, это немного лучше? Python версии 2.7

Answer 1

Если я вас правильно понял, вы могли бы сделать это следующим образом:

my_slice = slice(np.argwhere(x>5.5e3)[0], np.argwhere(x>6e3)[0]) 
x = data[my_slice,0] 
y = data[my_slice,1] 

np.argwhere(x>5.5e3)[0] является индекс первого вхождения x>5.5e3 и как мудр для конца среза. (Предполагая, что ваши данные сортируются)

Более общий способ работает, даже если ваши данные не сортируется:

mask = (x>5.5e3) & (x<6e3) 
x = data[mask, 0] 
y = data[mask, 1] 

Answer 2

решена с помощью

plt.axis([5500, 6000, 0, 8e-15]) 

спасибо за помощь.