| import os | |
| def save_dataframes(dataframes, target, save_path): | |
| """ | |
| Save pandas DataFrames to CSV files. | |
| Args: | |
| dataframes (dict): Dictionary mapping dataset names to DataFrames | |
| target (str): Target dataset name to save, or 'all' to save all | |
| save_path (str): Path to save the CSV files | |
| """ | |
| os.makedirs(save_path, exist_ok=True) | |
| if target == "all": | |
| for name, df in dataframes.items(): | |
| df.to_csv(os.path.join(save_path, f"{name}.csv"), index=False) | |
| elif target in dataframes: | |
| dataframes[target].to_csv(os.path.join(save_path, f"{target}.csv"), index=False) | |
| else: | |
| raise ValueError(f"Unknown target: {target}") | |