将 for 循环的输出存储在数组中

Question

我的 Python 之旅还处于起步阶段，所以这个问题对于更高级的程序员来说可能是基础。

我想分析一堆 .wav 文件，它们都存储在同一个目录中，所以我创建了一个所有文件名的列表，这样我就可以得到它们的音频信号和采样率。

dirPath=r"path_to_directory"
global files
files = [f for f in os.listdir(dirPath) if os.path.isfile(os.path.join(dirPath, f)) and f.endswith(".wav")]
for file_name in files:
    path_to_file= dirPath+"\\"+file_name
    audio_signal, sample_rate = sf.read(path_to_file)

sf是soundfile库。

audio_signal是一个数组， sample_rate是一个数字。

现在我希望能够将audio_signal和sample_rate与相应的file_name一起存储，以便我以后可以访问它们。 我该怎么做呢？

我试过了

files = [f for f in os.listdir(dirPath) if os.path.isfile(os.path.join(dirPath, f)) and f.endswith(".wav")],[]
for file_name in files[0]:
    path_to_file= dirPath+"\\"+file_name
    audio_signal, sample_rate = sf.read(path_to_file)
    files[1].append(audio_signal)
    files[2].append(sample_rate)

这似乎有效，但有没有更优雅的方式？ 我觉得audio_signal 、 sample_rate和file_name是单独的值而不是相互依赖的。

Answer 1

您要查找的数据结构是一个关联数组，它将键与值相关联——键是文件名，值是本例中由音频信号和采样率组成的元组。

关联数组的实现作为字典类型内置于 Python 中。

您可以在此处了解词典： https : //docs.python.org/3/tutorial/datastructures.html#dictionaries

您代码中的应用程序如下所示：

result = {}

for file_name in files:
    # as before:
    # ...
    audio_signal, sample_rate = ...
    # new:
    result[file_name] = audio_signal, sample_rate