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

Question

我正在遍历目录中的文件，并希望将文件名和我从文件中提取的一些内容保存在同一个 pandas 数据框中。在浏览目录中的所有文件时，如何将 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

Answer 1

你需要这个功能吗？

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

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

Iterating through txt files in directory, saving filenames

python

glob

pandas

txt