diff --git a/electrode_check.py b/electrode_check.py index 9d743df..ac4ebcd 100644 --- a/electrode_check.py +++ b/electrode_check.py @@ -362,7 +362,7 @@ def main(): ylim = (yspan[0] - np.abs(np.diff(yspan)) * 0.2, yspan[1] + np.abs(np.diff(yspan)) * 0.2) for ch in channel_array: - h, = Plot.axs[ch].plot(np.arange(250)[:len(channel_data[ch])] / rate, channel_data[ch] / 1000, color='k') + h, = Plot.axs[ch].plot(np.arange(250)[:len(channel_data[ch])] / rate, np.array(channel_data[ch]) / 1000, color='k') Plot.axs[ch].set_ylim(ylim) Plot.channel_handle.append(h) @@ -373,7 +373,7 @@ def main(): yspan = [np.min(channel_data[power_channel]), np.max(channel_data[power_channel])] ylim = [yspan[0] - np.abs(np.diff(yspan)) * 0.2, yspan[1] + np.abs(np.diff(yspan)) * 0.2] for ch in channel_array: - Plot.channel_handle[ch].set_data(np.arange(250)[:len(channel_data[ch])] / rate, channel_data[ch] / 1000) + Plot.channel_handle[ch].set_data(np.arange(250)[:len(channel_data[ch])] / rate, np.array(channel_data[ch]) / 1000) Plot.axs[ch].set_ylim(ylim) Plot.fig.canvas.draw()