首页 > 解决方案 > 计算时代的平均值

问题描述

我正在使用 MNE Toolbox 对 Python 进行脑电图数据分析。我是一个 Python 新手,我想知道是否有办法计算一个时代的平均值?“纪元平均值”是指每一个纪元,并找到它会给出的平均曲线。(英语不是我的第一语言,所以我希望很清楚)谢谢你的帮助!

标签: pythonmne-python

解决方案


假设每个 epoch 总共有 100 个数据点,而你有这样的 20 个 epoch。然后,您可以将这些数据重塑为(20,100):20 行和 100 列。你想要的是每个时代的平均值。我假设您不需要滚动平均值(移动平均线:MA);如果您需要 MA,请在评论部分留言。

让我们制作一些虚拟数据并将此逻辑应用于它。

import numpy as np
import matplotlib.pyplot as plt

%matplotlib inline
%config InlineBackend.figure_format = 'svg' # 'svg', 'retina'
plt.style.use('seaborn-white')

x = np.random.randn(20*100) # assume this is your original data
x = x.reshape((20,100)) # now we reshape it: here each row is an epoch
epoch_average = x.mean(axis=1)

# Plot figure to show results
show_figure = True
if show_figure:
    fig, axs = plt.subplots(nrows=5, ncols=4, figsize=(12,15), sharey='row')
    for i, (ax, x_epoch) in enumerate(zip(axs.flatten(), x)):
        plt.sca(ax)
        plt.plot(np.arange(x.shape[1]), x_epoch, 'k-', label='epoch-{}'.format(i))
        plt.axhline(epoch_average[i], label='epoch-average', color='red', alpha=0.8, lw=2.0, ls='--')
        plt.legend()
        plt.title('Epoch-{} Average: {:.3f}'.format(str(i).zfill(2), epoch_average[i]))

    plt.tight_layout()
    plt.show()
    fig.savefig('output.png', dpi=300)    

在此处输入图像描述


推荐阅读