将特定文件夹中的所有文件名写入 Pandas Dataframe 表列

Question

假设我有以下文件夹，其中包含 .txt 和 .wav 音频文件。

我想创建一个数据框表，其中每行数据都是这些 .wav 文件之一。作为示例输出，它看起来像这样：

总共应该有 460 行数据，每个 .wav 文件一行。

如何在 Python 代码中执行此操作？我已经在 StackOverflow 上进行了搜索，但找不到这样的示例。

Answer 1

您可以简单地使用

os.listdir

获取文件列表，并使用条件循环遍历它们以检查扩展名。

 import pandas as pd
 import os
 file_list = []
 for file in os.listdir('/path/to/dir'):
    if file.endswith('.wav'):
        file_list.append(file)
 df = pd.DataFrame({'Filename': file_list})

一旦获得文件名，您就可以按照您喜欢的方式将它们添加到 DataFrame 中。