时间戳

首页 > 解决方案 > 将 for 循环的输出存储在数组中

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

问题描述

我的 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不是相互依存的。

标签： pythonarraysappend

解决方案

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

关联数组的实现作为字典类型内置在 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

推荐阅读