reading nifti(or supported format) with nibabel

reading_nifti.py

import nibabel as nib

def read_file(filename):
  """
  Args:
  -----
    filename(str): path of your file
    
  Output:
  -------
    ndarray: numpy array of the data of the file
  """
  img = nib.load(filename)
  return img.get_data()