| import os | |
| import matplotlib.pyplot as plt | |
| def color_print(content): | |
| print(f'\033[1;46m{content}\033[0m\n') | |
| def save_pic_iterly(pic_name, postfix, info): | |
| pic_idx=1 | |
| pic_name_full=f'{pic_name}_{pic_idx}.{postfix}' | |
| while os.path.exists(pic_name_full): | |
| print(f'File {pic_name_full} already exists.') | |
| pic_idx += 1 | |
| pic_name_full=f'{pic_name}_{pic_idx}.png' | |
| plt.savefig(pic_name_full, dpi=300, bbox_inches='tight') | |
| color_print(f'!!!!! {info} is saved in file {pic_name_full}') |