Update python script to add padding on csv filenames

zeddo123 · May 7, 2023 · efcf5e3 · efcf5e3
1 parent 2545fc4
commit efcf5e3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/generate_set.py b/generate_set.py
@@ -17,11 +17,16 @@
 if not os.path.exists("csvset"):
     os.makedirs("csvset")
 
+padding_file = len(str(NUMBER_OF_FILES))
+padding_line = len(str(NUMBER_OF_LINES))
+
 for i in tqdm(range(NUMBER_OF_FILES)):
-    with open(f'csvset/{i}.csv', 'w') as f:
+    filename = str(i).zfill(padding_file)
+    with open(f'csvset/{filename}.csv', 'w') as f:
         uppercase = list(ascii_uppercase[:cols])
         f.write(','.join(uppercase))
         f.write('\n')
         for l in range(NUMBER_OF_LINES):
-            f.write(','.join(f'{letter}{l}' for letter in uppercase))
+            l_str = str(l).zfill(padding_line)
+            f.write(','.join(f'{filename}{letter}{l_str}' for letter in uppercase))
             f.write('\n')