【预备知识】使用os.makedirs()创建目录时exist_ok=True的作用

2.2.1. 读取数据集

举一个例子，我们首先创建一个人工数据集，并存储在CSV（逗号分隔值）文件 ../data/house_tiny.csv中。 以其他格式存储的数据也可以通过类似的方式进行处理。 下面我们将数据集按行写入CSV文件中。

import os

os.makedirs(os.path.join('..', 'data'), exist_ok=True)
data_file = os.path.join('..', 'data', 'house_tiny.csv')
with open(data_file, 'w') as f:
    f.write('NumRooms,Alley,Price\n')  # 列名
    f.write('NA,Pave,127500\n')  # 每行表示一个数据样本
    f.write('2,NA,106000\n')
    f.write('4,NA,178100\n')
    f.write('NA,NA,140000\n')

要从创建的CSV文件中加载原始数据集，我们导入pandas包并调用read_csv函数。该数据集有四行三列。其中每行描述了房间数量（“NumRooms”）、巷子类型（“Alley”）和房屋价格（“Price”）。

# 如果没有安装pandas，只需取消对以下行的注释来安装pandas
# !pip install pandas
import pandas as pd

data = pd.read_csv(data_file)
print(data)

   NumRooms Alley   Price
0       NaN  Pave  127500
1       2.0   NaN  106000
2       4.0   NaN  178100
3       NaN   NaN  140000

1、os.makedirs(os.path.join('..', 'data'), exist_ok=True)中exist_ok=True 选项的作用：

• 如果目录已经存在，不会报 FileExistsError。

• 如果目录不存在，则正常创建。

• 如果目录存在但不是一个文件，仍然不会报错。

• 如果目录存在且是一个文件，仍然会报错。

1. 默认情况下（exist_ok=False），如果目录已存在会报错：

import os

# 先创建一个目录
os.makedirs('example_dir')

# 再次尝试创建相同目录（默认 exist_ok=False）
os.makedirs('example_dir')  # 这里会报 FileExistsError

2. 使用 exist_ok=True，避免报错：

import os

# 第一次创建
os.makedirs('example_dir', exist_ok=True)

# 第二次尝试创建相同目录
os.makedirs('example_dir', exist_ok=True)  # 这次不会报错

3. 创建多级目录：

os.makedirs(os.path.join('..', 'data', 'subdir'), exist_ok=True)

这将创建 ../data/subdir 目录，如果 ../data 目录不存在，makedirs 也会递归创建它。

总结：

exist_ok=True 让 os.makedirs 在 目标目录已存在时不会报错，避免手动检查目录是否存在的额外代码，简化逻辑。