Python小功能汇总

 

        import os
        # 如果文件夹不存在，则创建
        save_dir = "train_model"  # 文件夹名称
        save_path = os.getcwd() + os.sep + save_dir
        if not os.path.isdir(save_path):
            os.makedirs(save_path)

　　

1、没有文件夹就新建

适用以下3种情况。

（1）文件夹适用

（2）相对路径适用

（3）绝对路径适用

 

    # 判断输出文件夹是否存在。不存在就创建
    # 1、output_dir为绝对路径
    if os.path.isabs(output_dir):
        if not os.path.isdir(output_dir):
            os.makedirs(output_dir)
    # 2、output_dir为相对路径（./out）或者文件夹（out）
    else:
        mk_dir = output_dir  # 文件夹名称
        mk_path = os.getcwd() + os.sep + mk_dir
        if not os.path.isdir(mk_path):
            os.makedirs(mk_path)