feat(script): Add zero-padding to CSV filenames to include sensors number

This commit is contained in:
nuluh
2024-08-27 10:11:39 +07:00
parent 8a499a04fb
commit b2684c23f6

View File

@@ -16,8 +16,10 @@ os.makedirs(processed_path, exist_ok=True)
# Define the number of zeros to pad # Define the number of zeros to pad
num_damages = 5 num_damages = 5
num_tests = 10 num_tests = 10
num_sensors = 2
damage_pad = len(str(num_damages)) damage_pad = len(str(num_damages))
test_pad = len(str(num_tests)) 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 for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1
damage_folder = f"DAMAGE_{damage:0{damage_pad}}" damage_folder = f"DAMAGE_{damage:0{damage_pad}}"
@@ -25,23 +27,24 @@ for damage in range(1, num_damages + 1): # 5 Damage levels starts from 1
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 for sensor in range(1, 3): # 2 Sensors per test
csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}.csv" # Filename for the CSV
csv_path = os.path.join(damage_path, csv_filename) 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 # Generate dummy data
num_rows = 10 num_rows = 10
start_time = datetime.now() start_time = datetime.now()
timestamps = [start_time + timedelta(seconds=i*0.0078125) for i in range(num_rows)] timestamps = [start_time + timedelta(seconds=i*0.0078125) for i in range(num_rows)]
values = np.random.randn(num_rows) # Random float values values = np.random.randn(num_rows) # Random float values
# Create DataFrame # Create DataFrame
df = pd.DataFrame({ df = pd.DataFrame({
"Time": timestamps, "Time": timestamps,
"Value": values "Value": values
}) })
# Save the CSV file with a custom header # Save the CSV file with a custom header
with open(csv_path, 'w') as file: with open(csv_path, 'w') as file:
file.write('sep=,\n') # Writing the separator hint file.write('sep=,\n') # Writing the separator hint
df.to_csv(file, index=False) df.to_csv(file, index=False)