15 lines
538 B
Python
15 lines
538 B
Python
import numpy as np
|
|
from scipy.signal import find_peaks
|
|
|
|
from lightkurve.periodogram import Periodogram
|
|
from lightkurve.lightcurve import LightCurve
|
|
|
|
def findMaxIndices(lc, num=3, distance=100):
|
|
if(isinstance(lc, Periodogram)):
|
|
data = lc.power
|
|
elif(isinstance(lc, LightCurve)):
|
|
data = lc.flux
|
|
peak_indices, peak_dict = find_peaks(data, height=(None, None), distance=distance)
|
|
peak_heights = peak_dict["peak_heights"]
|
|
maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
|
|
return maxIndices |