使用python绘制文件夹中所有.wav文件的分布图(在jupyter notebook中)

import os
import matplotlib.pyplot as plt

def get_wav_files(folder_path):
    wav_files = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(".wav"):
                wav_files.append(os.path.join(root, file))
    return wav_files

def plot_wav_files_distribution(folder_path):
    wav_files = get_wav_files(folder_path)
    file_count = len(wav_files)
    labels = [os.path.basename(file) for file in wav_files]
    sizes = [1] * file_count  # 设置每个文件的大小为1，可以根据需要进行调整

    plt.figure(figsize=(10, 6))  # 设置图形的大小

    # 绘制饼图
    plt.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90)

    # 添加标题
    plt.title("Distribution of .wav Files")

    # 显示图形
    plt.show()

folder_path = "your_folder_path"
plot_wav_files_distribution(folder_path)