python - 根据 csv 文件名命名数据框？

Question

试图批量分析一个充满.csv 文件的文件夹，然后根据.csv 名称再次将它们保存出来。但是，我无法仅提取文件名并将其分配给数据框 (df)。

import glob
import pandas as pd

path = r'csv_in'
allFiles = glob.glob(path + '/*.csv')

for file_ in allFiles:   
    df = pd.read_csv(file_, header=0)
    df.name = file_
    print(df.name)

我得到的打印结果是“csv_in/*.csv”。

我要查找的结果只是 csv 名称“*.csv”

Answer 1

使用 [] 创建新列，使用 os.path.normpath 和 os.path.basename 创建新列:

import os

for file_ in allFiles:   
    df = pd.read_csv(file_, header=0)
    df['name'] = os.path.basename(os.path.normpath(file_))
    #if need remove extension (csv)
    #df['name'] = os.path.splitext(os.path.basename("hemanth.txt"))[0]
    print(df.name)