8.python wave 批量截取2s音頻

# @File  : sound_cut.py
# @Author: Wang Zhimin
# @Date  : 2019/10/12
import os
import wave
import numpy as np
import pylab as plt
def load_wav(path):
    """
        功能：讀取檔案夾下wav檔案，進行2s剪切
        輸入：檔案夾路徑
        輸出：歸一化後的資料
        時間：2019年10月12日
        版本：V1.0
    """

    files = []

    # 讀檔案

    for f in os.listdir(path):
        if not f.startswith('.') and f.endswith('.wav'):
            files.append(f)
    # 多通道處理，以及采樣頻率歸一化
    # print(files)
    for f in files:
        w = wave.open(path+"/"+f, "rb")
        params = w.getparams()
        nchannels, sampwidth, framerate, nframes = params[:4]
        nums=int(nframes / (framerate * 2)) + 1
        # print("片段個數:",nums)
        # print("framerate",framerate)
        # print("nchannels", nchannels)
        # print("sampwidth", sampwidth)
        # print("nframes", nframes)

        str_data = w.readframes(nframes)
        wave_data = np.fromstring(str_data, dtype=np.short)
        for i in np.arange(nums):
            print((i * 2) * framerate, (i * 2 + 2) * framerate)
            # str_data1 = wave_data[0:2*framerate]
            str_data1 = wave_data[(i * 2) * framerate:(i * 2 + 2) * framerate]
            w.close()

            # 打開WAV文檔
            f1 = wave.open(r"D:/work/音頻剪切原始資料/"  +str(i+1)+"-"+ f, "wb")

            # 配置聲道數、量化位數和取樣頻率
            f1.setnchannels(nchannels)
            f1.setsampwidth(sampwidth)
            f1.setframerate(framerate)
            # 将wav_data轉換為二進制資料寫入檔案
            f1.writeframes(str_data1.tostring())
        # print(len(str_data1))
        # f1.close()
        # print(f)
    return 0


if __name__ == '__main__' :
    load_wav("D:/work/音頻原始資料")