1d_ecg2_zeropad.py

def zero_pad(data, length):
    extended = np.zeros(length)
    siglength = np.min([length, data.shape[0]])
    extended[:siglength] = data[:siglength]
    return extended

# plot a sample
data = sio.loadmat('training2017/A00001.mat')['val'][0]

# 61 seconds is the maximum length in our dataset
data = zero_pad(data, max_length*freq)

plt.figure(figsize=(15, 5))
plt.title('Sample zero-padded ECG recording', fontsize=30)
plt.xlabel('Time (s)', fontsize=25)
plt.ylabel('Potential (mV)', fontsize=25)
plt.plot(np.arange(0, data.shape[0])/300, data/1000)