nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

feat(script): add zero-padding to CSV filenames and change the output generated csv as raw data in raw folder

Browse Source
This commit is contained in:
nuluh
2024-08-17 19:51:42 +07:00
parent 1c23edf098
commit 553140fe3c
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
generate_dummy_data.py
View File

@@ -13,14 +13,20 @@ processed_path = os.path.join(base_path, "processed")
os.makedirs(raw_path, exist_ok=True) os.makedirs(raw_path, exist_ok=True)
os.makedirs(processed_path, exist_ok=True) os.makedirs(processed_path, exist_ok=True)
for damage in range(1, 6): # 5 Damage levels # Define the number of zeros to pad
damage_folder = f"DAMAGE_{damage}" num_damages = 5
damage_path = os.path.join(processed_path, damage_folder) num_tests = 10
damage_pad = len(str(num_damages))
test_pad = len(str(num_tests))
for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1
damage_folder = f"DAMAGE_{damage:0{damage_pad}}"
damage_path = os.path.join(raw_path, damage_folder)
os.makedirs(damage_path, exist_ok=True) os.makedirs(damage_path, exist_ok=True)
for test in range(1, 11): # 10 Tests per damage level for test in range(1, 11): # 10 Tests per damage level
# Filename for the CSV # Filename for the CSV
csv_filename = f"D{damage}_TEST{test}.csv" csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}.csv"
csv_path = os.path.join(damage_path, csv_filename) csv_path = os.path.join(damage_path, csv_filename)
# Generate dummy data # Generate dummy data