make_data_body.py

import os
import csv

data_folder = "data/body/data_body"
output_csv = "data_body_12classes_24112023.csv"

combined_data = []
subfolder_index = 0  # Initialize subfolder index

for subdir, _, files in os.walk(data_folder):
    if subdir != data_folder:  # Skip the root folder
        for file in files:
            if file.endswith(".txt"):
                with open(os.path.join(subdir, file), "r") as txt_file:
                    lines = txt_file.readlines()
                    for line in lines[1:]:
                        values = line.strip().split(",")
                        combined_data.append([str(subfolder_index)] + values[1:])
        subfolder_index += 1

with open(output_csv, "w", newline="") as csvfile:
    csv_writer = csv.writer(csvfile)
    csv_writer.writerows(combined_data)