基于Python实现录音功能的示例代码

我来为您讲解一下“基于Python实现录音功能的示例代码”的完整攻略。

1. 安装必要的库

在Python中实现录音功能，需要用到pyaudio库。如果还没有安装过这个库，可以通过以下命令进行安装：

pip3 install pyaudio

2. 编写代码

下面是一个简单的示例，展示如何使用pyaudio库实现录音功能。

import pyaudio
import wave

# 设置录音参数
CHUNK = 1024
FORMAT = pyaudio.paInt16
CHANNELS = 1
RATE = 16000
RECORD_SECONDS = 5

# 录音
p = pyaudio.PyAudio()
stream = p.open(format=FORMAT,
                channels=CHANNELS,
                rate=RATE,
                input=True,
                frames_per_buffer=CHUNK)

print("开始录音......")

frames = []
for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
    data = stream.read(CHUNK)
    frames.append(data)

print("录音结束！")

# 存储录音文件
wf = wave.open("output.wav", 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
wf.writeframes(b''.join(frames))
wf.close()

# 停止录音
stream.stop_stream()
stream.close()
p.terminate()

上面的代码首先设置了录音参数，包括采样率、通道数、录制时长等等。然后通过pyaudio库获取到音频输入设备，并开始录音。录音结束后，将录制的音频保存为WAV文件，并停止录音。

3. 示例说明

示例1

上面的示例代码适用于对音频质量要求不高的简单应用场景。如果需要录制高品质的音频，可以对参数进行调整，比如增加采样率和位深度，减少数据块大小等。

示例2

如果需要在录制过程中实时显示波形图，可以通过matplotlib库来实现。具体做法是在录音的过程中，将采集到的音频数据保存到numpy数组中，然后根据这些数据绘制波形图。下面是代码示例：

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

# 设置录音参数
CHUNK = 1024
FORMAT = pyaudio.paInt16
CHANNELS = 1
RATE = 16000
RECORD_SECONDS = 5

# 录音
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), '-')
ax.set_ylim(-32768, 32767)
ax.set_xlabel('Time')
ax.set_ylabel('Amplitude')

print("开始录音......")

frames = []
for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
    data = stream.read(CHUNK)
    frames.append(data)
    y = np.frombuffer(data, dtype=np.int16)
    line.set_ydata(y)
    fig.canvas.draw()
    fig.canvas.flush_events()

print("录音结束！")

# 存储录音文件
wf = wave.open("output.wav", 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
wf.writeframes(b''.join(frames))
wf.close()

# 停止录音
stream.stop_stream()
stream.close()
p.terminate()

上述代码通过matplotlib库绘制波形图，并在录音过程中实时更新图像。需要注意的是，为了使波形图能实时显示，要使用canvas.draw()来刷新绘图窗口，并在每次更新完之后，调用flush_events()方法来刷新事件队列，否则，图像将不能实时更新。