feat(script): Add zero-padding to CSV filenames to include sensors number
This commit is contained in:
@@ -16,8 +16,10 @@ os.makedirs(processed_path, exist_ok=True)
|
||||
# Define the number of zeros to pad
|
||||
num_damages = 5
|
||||
num_tests = 10
|
||||
num_sensors = 2
|
||||
damage_pad = len(str(num_damages))
|
||||
test_pad = len(str(num_tests))
|
||||
sensor_pad = len(str(num_sensors))
|
||||
|
||||
for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1
|
||||
damage_folder = f"DAMAGE_{damage:0{damage_pad}}"
|
||||
@@ -25,8 +27,9 @@ for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1
|
||||
os.makedirs(damage_path, exist_ok=True)
|
||||
|
||||
for test in range(1, 11): # 10 Tests per damage level
|
||||
for sensor in range(1, 3): # 2 Sensors per test
|
||||
# Filename for the CSV
|
||||
csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}.csv"
|
||||
csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}_{sensor:0{sensor_pad}}.csv"
|
||||
csv_path = os.path.join(damage_path, csv_filename)
|
||||
|
||||
# Generate dummy data
|
||||
|
||||
Reference in New Issue
Block a user