Нужно изменить функцию для получения матрицы подзаговоров - PullRequest
Новая
       7

Нужно изменить функцию для получения матрицы подзаговоров

0 голосов
/ 27 марта 2019

Я использую следующую функцию обнаружения пиков в моем коде: https://nbviewer.jupyter.org/github/demotu/BMC/blob/master/notebooks/DetectPeaks.ipynb Последние 20 строк или около того этой функции относятся к графику, который выполняется автоматически с аргументом show = True. У меня есть 18 файлов данных, и я хотел бы изменить эту функцию (при необходимости), чтобы получать 6x3 вспомогательных участков графиков обнаружения пиков, аналогичных приведенным в ссылке, вместо того, чтобы создавать их по отдельности.

Вот что я уже пробовал: 

fig, axes = plt.subplots(nrows=6, ncols=3, figsize=(20,20))
for i, ax in enumerate(axes.flatten()):
    for idx, file in enumerate(files):
        if i == idx:
            detect_peaks(P[file], show=True)

Вот часть функции, которую мне, возможно, потребуется изменить: 

def _plot(x, mph, mpd, threshold, edge, valley, ax, ind):
    """Plot results of the detect_peaks function, see its help."""
    try:
        import matplotlib.pyplot as plt
    except ImportError:
        print('matplotlib is not available.')
    else:
        if ax is None:
            _, ax = plt.subplots(1, 1, figsize=(8, 4))

        ax.plot(x, 'b', lw=1)
        if ind.size:
            label = 'valley' if valley else 'peak'
            label = label + 's' if ind.size > 1 else label
            ax.plot(ind, x[ind], '+', mfc=None, mec='r', mew=2, ms=8,
                    label='%d %s' % (ind.size, label))
            ax.legend(loc='best', framealpha=.5, numpoints=1)
        ax.set_xlim(-.02*x.size, x.size*1.02-1)
        ymin, ymax = x[np.isfinite(x)].min(), x[np.isfinite(x)].max()
        yrange = ymax - ymin if ymax > ymin else 1
        ax.set_ylim(ymin - 0.1*yrange, ymax + 0.1*yrange)
        ax.set_xlabel('Data #', fontsize=14)
        ax.set_ylabel('Amplitude', fontsize=14)
        mode = 'Valley detection' if valley else 'Peak detection'
        ax.set_title("%s (mph=%s, mpd=%d, threshold=%s, edge='%s')"
                     % (mode, str(mph), mpd, str(threshold), edge))
        # plt.grid()
        plt.show()

В результате получается пустая матрица субплотов 6x3, где каждый график обнаружения пиков отображается отдельно под ней.

Я ищу что-то вроде этого:

https://imgur.com/U46eF3e

...