python声音频谱分析代码

这里是一个使用pyaudio库进行实时音频频谱分析的简单示例代码：

import pyaudio
import numpy as np
import matplotlib.pyplot as plt

# 设置参数
CHUNK = 1024  # 每个缓冲区的大小
FORMAT = pyaudio.paInt16  # 采样格式
CHANNELS = 1  # 声道数
RATE = 44100  # 采样率

# 初始化pyaudio对象
p = pyaudio.PyAudio()

# 打开音频流
stream = p.open(format=FORMAT,
                channels=CHANNELS,
                rate=RATE,
                input=True,
                frames_per_buffer=CHUNK)

# 绘制频谱图
fig, ax = plt.subplots()
x = np.arange(0, 2 * CHUNK, 2)
line, = ax.plot(x, np.random.rand(CHUNK), '-', lw=2)
ax.set_ylim(0, 255)
ax.set_xlim(0, CHUNK)
plt.xlabel('Frequency')
plt.ylabel('Amplitude')

# 实时更新频谱图
while True:
    data = stream.read(CHUNK)
    data = np.frombuffer(data, dtype=np.int16)
    fft = np.fft.fft(data)
    fft_abs = np.abs(fft)[:CHUNK]
    line.set_ydata(fft_abs)
    fig.canvas.draw()
    fig.canvas.flush_events()

# 关闭音频流和pyaudio对象
stream.stop_stream()
stream.close()
p.terminate()

在上述代码中，首先设置了一些参数，例如缓冲区大小、采样格式、声道数、采样率等。然后初始化了一个pyaudio对象，并打开了一个音频流，用于实时读取音频数据。接着绘制了一个空的频谱图，其中x轴表示频率，y轴表示振幅。最后使用一个while循环，不断读取音频数据并进行FFT变换，得到音频的频谱信息，然后实时更新频谱图。可以根据需要对频谱图进行调整和美化，例如修改颜色、坐标轴刻度、标题等。最后在合适的时候关闭音频流和pyaudio对象。