遍历目录中的txt文件,保存文件名

问题描述

我正在遍历目录中的文件,并希望将文件名和从文件提取的一些内容保存在同一只熊猫数据框中。在浏览目录中的所有文件时,如何将txt文件名称保存在列表中(然后将其作为单独的列插入到pandas数据框中)?

这是我的代码的一部分:

columns_df = ['file','stuff']
df_stuff = pd.DataFrame(columns = columns_df)

filenamelist = []
stufflist = []

os.chdir(r'path\to\directory')
for file in glob.glob('*.txt'):
    # Extract some stuff from file and append to stufflist (DONE)
    # Save filename in the filenamelist (THE PROBLEM)

df_stuff['stuff'] = stufflist
df_stuff['file'] = filenamelist

解决方法

您需要此功能吗?

for file in glob.glob('*.txt'):
    filenamelist.append(file)