python 「wavファイルのリストを作成する」方法を教えてください。

Question

「wavファイルのリストを作成する」方法を教えてください。

wavファイルを読み込み、スペクトログラムを表示し、スペクトログラムのデータをcsvで保存するコードをまず私が書き、
それを先生に加工してもらい、データのリストをfor文でなぞっていくコードにしてもらいました。
以下がそのコードです。

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import print_function
import librosa
import librosa.display
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

#DataList(wavファイルのリストを作成する)
InDataList = pd.read_csv("InDataList.csv")
OutDataList = pd.read_csv("OutDataList.csv")

for i in len(DataList):
    # ファイル読込
    y, sr = librosa.load(DataList[i])

    # ファイル読込
    #y, sr = librosa.load('BRAVE JEWEL.wav')

    # 短時間フーリエ変換
    S = np.abs(librosa.stft(y))

    # 画像出力
    plt.figure(figsize=(10,6))
    librosa.display.specshow(librosa.amplitude_to_db(S, ref=np.max), y_axis='log', x_axis='time')
    plt.title('Power spectrogram')
    plt.colorbar(format='%+2.0f dB')

     #テラテイルよりアドバイス
    #plt.yscale('linear')  # 線形スケールに設定
    #plt.yticks(np.arange(0, 10001, 1000))  # 目盛りを [0, 1000, 2000, ..., 10000] にする。

    plt.tight_layout()
    plt.show()

    #Axesを描画する領域(Figure)を作成
    fig = plt.figure(facecolor = "lightgray")


    #https://jonkara.net/2018/10/07/librosa/
    #https://python.atelierkobato.com/matplotlib/

    ##np.savetxt("ab.csv", S, delimiter=",")
    #np.savetxt("BRAVE JEWEL.csv", S, fmt="%.1f",delimiter=",")
    ##%0.5fだと小数点以下５桁まで出力

    #データ出力
    np.savetxt(OutDataList[i], S, fmt="%.1f",delimiter=",")
    #%0.5fだと小数点以下５桁まで出力

11行目から、

#DataList(wavファイルのリストを作成する)
InDataList = pd.read_csv("InDataList.csv")
OutDataList = pd.read_csv("OutDataList.csv")

と書いてあるのですが、これはあらかじめどのようなファイル、データを作っておけばよいのでしょうか？

プログラミングに関して、まだ知らないことばかりなので質問自体にも不備があることと思いますが、どうかよろしくお願いします。

Answer 1

書き込むデータの形式としては

InDataList.csv

１番目に読み込むファイル名
２番目に読み込むファイル名
３番目に読み込むファイル名
・・・

OutDataList.csv

１番目に書き込むファイル名
２番目に書き込むファイル名
３番目に書き込むファイル名
・・・

のような形式かと思うのですが・・・

残念ながら加工してもらった部分にはバグがあるため動作しないと思います。
たぶんこういう事をやりたかったのだろうと想定して修正すると、以下のようになります。

#（略）

#DataList(wavファイルのリストを作成する)
# 修正（データをインデックスで読み込むにはこのような加工）
InDataList = pd.read_csv("InDataList.csv", header=None).iloc[:,0]
OutDataList = pd.read_csv("OutDataList.csv", header=None).iloc[:,0]

for i in range(len(InDataList)):　#<- 修正：変数名が間違い & range()追加
    # ファイル読込
    y, sr = librosa.load(InDataList[i]) #<- 修正：変数名が間違い

    #(中略)

    #データ出力
    np.savetxt(OutDataList[i], S, fmt="%.1f",delimiter=",")
    #%0.5fだと小数点以下５桁まで出力

---

ただ、個人的には入力・出力のファイルを２つ用意するよりは、１つにまとめて

DataList.csv

INPUTFILE,OUTPUTFILE
１番目に読み込むファイル名,１番目に書き込むファイル名
２番目に読み込むファイル名,２番目に書き込むファイル名
３番目に読み込むファイル名,２番目に書き込むファイル名
・・・

のようなデータ形式にして

#（略）

#DataList(wavファイルのリストを作成する)
DataList = pd.read_csv("DataList.csv")

for i, data in DataList.iterrows():
    # ファイル読込
    y, sr = librosa.load(data.INPUTFILE)

    #(中略)

    #データ出力
    np.savetxt(data.OUTPUTFILE, S, fmt="%.1f",delimiter=",")
    #%0.5fだと小数点以下５桁まで出力

のようにするのが良いかと思うのですが、そこはお好みで。。