用当前子目录名填充dataframe中的列值

import pandas as pd

# 读取数据集
df = pd.read_csv('dataset.csv')

# 创建新列
df['subdirectory'] = ''

import os

# 遍历每一行
for index, row in df.iterrows():
    # 获取当前子目录名
    subdirectory_name = os.path.basename(os.path.dirname(row['filepath']))
    # 填充到新列中
    df.at[index, 'subdirectory'] = subdirectory_name

# 查看更新后的dataframe
print(df)

# 将dataframe保存到新文件中
df.to_csv('updated_dataset.csv', index=False)