如何从路径字符串中获取文件夹名称，并将其添加到pandas dataframe中的新列中？

import os
import pandas as pd

def get_folder_name(path):
    folder_name = os.path.basename(os.path.dirname(path))
    return folder_name

data = {'path': ['path/to/file1', 'path/to/file2', 'path/to/file3']}
df = pd.DataFrame(data)

df['folder_name'] = df['path'].apply(get_folder_name)

print(df)

            path folder_name
0  path/to/file1        path
1  path/to/file2        path
2  path/to/file3        path