import nibabel as nib
import matplotlib.pyplot as plt
img = nib.load('group_7_34_18_Z.nii.gz')
img.shape

def show_slices(slices):
    """ Function to display row of image slices """
    fig, axes = plt.subplots(1, len(slices))
    for i, slice in enumerate(slices):
        axes[i].imshow(slice.T, cmap="gray", origin="lower")

slice_0 = img.get_fdata()[26, :, :]
slice_1 = img.get_fdata()[:, 40, :]
slice_2 = img.get_fdata()[:, :, 26]
show_slices([slice_0, slice_1, slice_2])
plt.suptitle("Center slices for EPI image")
plt.show()
