Skip to content

Commit efcf5e3

Browse files
committed
Update python script to add padding on csv filenames
1 parent 2545fc4 commit efcf5e3

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

generate_set.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,16 @@
1717
if not os.path.exists("csvset"):
1818
os.makedirs("csvset")
1919

20+
padding_file = len(str(NUMBER_OF_FILES))
21+
padding_line = len(str(NUMBER_OF_LINES))
22+
2023
for i in tqdm(range(NUMBER_OF_FILES)):
21-
with open(f'csvset/{i}.csv', 'w') as f:
24+
filename = str(i).zfill(padding_file)
25+
with open(f'csvset/{filename}.csv', 'w') as f:
2226
uppercase = list(ascii_uppercase[:cols])
2327
f.write(','.join(uppercase))
2428
f.write('\n')
2529
for l in range(NUMBER_OF_LINES):
26-
f.write(','.join(f'{letter}{l}' for letter in uppercase))
30+
l_str = str(l).zfill(padding_line)
31+
f.write(','.join(f'{filename}{letter}{l_str}' for letter in uppercase))
2732
f.write('\n')

0 commit comments

Comments
 (0)