EDF concatenated file stim channel fix using the .mat

edf_raw_fix.py

import numpy as np
import scipy as sp
import mne

mat = sp.io.loadmat('trig_s2.mat')
triggers, latency, urevent, duration, epoch = zip(*np.ravel(mat['triggers']))
latency = np.array(latency).ravel()
triggers = np.array(triggers, int).ravel()

raw = mne.fiff.edf.read_raw_edf(raw.info,'s2_2.edf', preload=True)
stim_channel = np.zeros(raw.last_samp+1)
stim_channel[latency] = triggers

picks = mne.fiff.pick_types(meg=False, stim=True)
raw._data[picks] = stim_channel

raw.save('s2_2_stim_fixed.fif', verbose=False)

raw_fixed = mne.fiff.Raw('s2_2_stim_fixed.fif')