dataset-builder / data1 /rename2.py
DouDou
Upload data1/rename2.py with huggingface_hub
e80a5c5 verified
import os
import glob
# get all target csv files
file_list = glob.glob("./datasets/all/chempile_code_complete_*.csv")
for file_path in file_list:
# extract filename and number part
dir_name, file_name = os.path.split(file_path)
prefix, num_suffix = file_name.split("_")[-2], file_name.split("_")[-1].split(".")[0]
# convert number to 3-digit format
new_num = num_suffix.zfill(3)
# construct new file path
new_file_name = file_name.replace(f"{prefix}_{num_suffix}", f"{prefix}_{new_num}")
new_file_path = os.path.join(dir_name, new_file_name)
# rename file
os.rename(file_path, new_file_path)