MNE interpolate new scalp channels

mne_interpolate.py

import mne

mneraw = mne.read_raw_edf('file.edf')
electrodes = ['Fpz', 'Cz']

# if electrodes to plot are not available, try to interpolate them
electrodes_not_in_list = [e for e in electrodes if e not in mneraw.ch_names]

if len(electrodes_not_in_list) > 0:
    try:
        mneraw = mne.add_reference_channels(mneraw, ref_channels=electrodes_not_in_list, copy=False)
        scalpmontage = mne.channels.make_standard_montage('standard_1020')
        mneraw = mneraw.set_montage(None)  # remove montage
        mneraw = mneraw.set_montage(scalpmontage, on_missing='warn')  # re-add montage for Fpz electrode

        # for e in electrodes_not_in_list:
        mneraw.info['bads'].extend(electrodes_not_in_list)
        mneraw = mneraw.interpolate_bads(reset_bads=True)

    except Exception as ex:
        logging.info('Could not add ' + repr(electrodes_not_in_list) + ' and interpolate: ' + repr(ex))