如何将Python实时麦克风频谱图集成到自定义PyQt5 GUI？

核心思路 #

把原Matplotlib独立窗口的绘图逻辑封装成PyQt兼容组件，嵌入到自定义GUI的指定区域，结合QThread+信号槽实现非阻塞式的音频数据读取与绘图更新，避免UI卡顿。

步骤1：封装实时FFT绘图组件 #

创建RealTimeFFTWidget.py，整合音频读取、FFT计算、绘图逻辑：

import matplotlib.pyplot as plt
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
import numpy as np
from PyQt5.QtCore import QThread, pyqtSignal
import soundcardlib  # 导入你的音频读取库

# 音频读取子线程（避免阻塞UI）
class AudioThread(QThread):
    data_ready = pyqtSignal(np.ndarray)

    def __init__(self, sample_rate=44100, chunk_size=1024):
        super().__init__()
        self.sample_rate = sample_rate
        self.chunk_size = chunk_size
        self.mic = soundcardlib.get_microphone()  # 匹配你的soundcardlib麦克风获取方法
        self._running = True

    def run(self):
        while self._running:
            # 读取音频数据（根据soundcardlib实际方法调整）
            raw_data = self.mic.record(numframes=self.chunk_size)
            # 多声道转单声道
            if len(raw_data.shape) > 1:
                raw_data = np.mean(raw_data, axis=1)
            self.data_ready.emit(raw_data)

    def stop(self):
        self._running = False
        self.wait()

# 实时绘图组件
class RealTimeFFTWidget(FigureCanvas):
    def __init__(self, parent=None, width=8, height=6, dpi=100):
        # 初始化Matplotlib画布
        self.fig = Figure(figsize=(width, height), dpi=dpi)
        self.ax_time = self.fig.add_subplot(211)
        self.ax_fft = self.fig.add_subplot(212)
        super().__init__(self.fig)
        self.setParent(parent)

        # 初始化参数
        self.sample_rate = 44100
        self.chunk_size = 1024
        self.time_axis = np.arange(self.chunk_size) / self.sample_rate
        self.fft_axis = np.linspace(0, self.sample_rate/2, self.chunk_size//2 + 1)

        # 初始化曲线
        self.line_time, = self.ax_time.plot(self.time_axis, np.zeros(self.chunk_size))
        self.line_fft, = self.ax_fft.plot(self.fft_axis, np.zeros(self.chunk_size//2 + 1))

        # 设置坐标轴
        self.ax_time.set(title="Time Domain Waveform", ylabel="Amplitude", xlabel="Time (s)")
        self.ax_fft.set(title="Frequency Spectrum", ylabel="Magnitude", xlabel="Frequency (Hz)")
        self.ax_time.set_ylim(-1, 1)
        self.ax_fft.set_ylim(0, 1)
        self.fig.tight_layout()

        # 启动音频线程与更新逻辑
        self.audio_thread = AudioThread(self.sample_rate, self.chunk_size)
        self.audio_thread.data_ready.connect(self.update_plot)
        self.audio_thread.start()

    def update_plot(self, audio_data):
        # 更新时域曲线
        self.line_time.set_ydata(audio_data)
        
        # 计算并更新频域曲线
        fft_vals = np.fft.rfft(audio_data)
        fft_mag = np.abs(fft_vals) / self.chunk_size
        self.line_fft.set_ydata(fft_mag)

        # 刷新画布（非阻塞）
        self.fig.canvas.draw_idle()

    def closeEvent(self, event):
        # 关闭时停止线程，释放资源
        self.audio_thread.stop()
        event.accept()

步骤2：整合到PyQt5 GUI逻辑 #

在你的GUI逻辑文件（比如gui_logic.py）中，将RealTimeFFTWidget添加到指定区域：

from PyQt5.QtWidgets import QMainWindow, QApplication, QVBoxLayout
from PyQt5.uic import loadUi
from RealTimeFFTWidget import RealTimeFFTWidget
import sys

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        # 加载UI文件
        loadUi("gui_easy.ui", self)

        # 将FFT组件添加到指定容器（假设UI中有名为fft_container的QWidget）
        self.fft_widget = RealTimeFFTWidget(parent=self.fft_container)
        # 给容器设置布局（如果UI中未预先设置）
        container_layout = self.fft_container.layout()
        if not container_layout:
            container_layout = QVBoxLayout(self.fft_container)
        container_layout.addWidget(self.fft_widget)

        # 可选：添加Matplotlib导航工具栏
        # from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
        # self.toolbar = NavigationToolbar(self.fft_widget, self.fft_container)
        # container_layout.addWidget(self.toolbar)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

关键注意事项 #

1. 匹配soundcardlib方法：确保AudioThread中调用的麦克风获取、音频录制方法与你的soundcardlib.py完全一致，比如如果你的库用get_default_mic()而非get_microphone()，要对应修改。
2. UI区域命名：确认UI文件中用于放置频谱图的组件对象名是fft_container，或修改代码中的对应名称。
3. 性能调优：根据设备性能调整chunk_size（如512/1024/2048），避免UI卡顿。
4. 资源释放：关闭窗口时通过closeEvent停止音频线程，避免内存泄漏。

内容的提问来源于stack exchange，提问作者M.Navidi