Блог

Моя цель — получить индексы локальных максимальных высот фрейма данных. Они варьируются от 3 до 5 для каждого столбца.

Я пробовал использовать функцию apply, но получаю либо сообщение об ошибке Shape of passed values is (736, 4), indices imply (736, 480) , либо could not broadcast input array from shape (0) into shape (1) .

Моя матрица 480x736 .

Вот что я написал для функции apply:

 import numpy as np
import peakutils
df.apply(lambda x: peakutils.indexes(x, thres=0.02/max(x), min_dist=100))
 

Вот что я могу заставить работать:

 indexes =[]
import numpy as np
import peakutils
for column in df:
    indexes.append(peakutils.indexes(df[column], thres=0.02/max(df[column]), min_dist=100))
 

Часто индексы имеют длину 4, но иногда я получаю на 1 больше или меньше:

 Out[32]: 
[array([ 12, 114, 217, 328, 433]),
 array([ 12, 116, 217, 325, 433]),
 array([ 64, 166, 283, 389]),
 array([105, 217, 326, 433]),
 array([105, 237, 390])]
 

Я предполагаю, что проблема с выводом возникает из-за того, что я не знаю форму результирующего фрейма данных. Форма не поддается определению с самого начала.

Как мне применить функцию к df, где выходные данные отличаются по размеру и типу?

pandas пытается сделать «что-то» с массивами. Вы можете замкнуть это «что-то», обернув lambda возвращаемое значение в pd.Series

Попробуйте это:

 df.apply(lambda x: pd.Series(peakutils.indexes(x, thres=0.02/max(x), min_dist=100)))