Вызов функции с apply_along_axis с несоответствием размера возвращаемого массива - PullRequest
Новая
       16

Вызов функции с apply_along_axis с несоответствием размера возвращаемого массива

0 голосов
/ 09 февраля 2020

Я вызываю функцию, которая возвращает пики, найденные во временном ряду. Входные данные (allAmplitude) - это 3-х мерный массив - изображения, сложенные друг на друга (1-й тусклый - номер изображения во времени, 2-й и 3-й - в пикселях) 

def mainFrequency(plot):
    peaks,_ = signal.find_peaks(plot,distance = 1)
    if len(peaks) > 1 :
        return peaks
    else:
        return int(10)

mFs = np.apply_along_axis(mainFrequency,0,allAmplitudes)

mainFrequency возвращает массивы разных размеров, основанные на входные данные (очевидно). Ошибка, которую я постоянно получаю, это несоответствие формы массива. Есть ли способ заставить это работать?

На данный момент функция mainFrequency не завершена, поэтому она не только обернута find_peaks в другую функцию

Заранее спасибо за Ваше время!

...