remove_data.py

#!/usr/bin/env python

# Load libraries.
import os, sys, shutil, argparse
from helper_code import *

# Parse arguments.
def get_parser():
    description = 'Remove data from the dataset.'
    parser = argparse.ArgumentParser(description=description)
    parser.add_argument('-i', '--input_folder', type=str, required=True)
    parser.add_argument('-o', '--output_folder', type=str, required=True)
    return parser

# Run script.
def run(args):
    # Iterate over subfolders in the input folder.
    for root, folder_names, file_names in os.walk(args.input_folder):
        input_folder = root

        # Extract the subfolder of the input folder and use it for the output folder.
        x, y = root, ''
        while not os.path.samefile(args.input_folder, x):
            x, y = os.path.split(x)[0], os.path.join(os.path.split(x)[1], y)
        output_folder = os.path.join(args.output_folder, y)
        
        # Create the output folder if it does not already exist.
        os.makedirs(output_folder, exist_ok=True)

        # Iterate over each file in each subfolder.
        for file_name in file_names:           
            input_file = os.path.join(input_folder, file_name)
            output_file = os.path.join(output_folder, file_name)

            parent_folder = os.path.basename(input_folder)
            file_root, file_ext = os.path.splitext(file_name)

            # If the file is not the binary signal data, then copy it.
            if not (file_ext == '.mat'):
                shutil.copy2(input_file, output_file)
            # Otherwise, if the file is signal data, then do not copy it.
            else:
                pass 

if __name__=='__main__':
    run(get_parser().parse_args(sys.argv[1:]))