python下麦克风设备选择和录音

import pyaudio
import threading
import wave 
import numpy  as np def audio_f2i(data, width=16):"""将浮点数音频数据转换为整数音频数据。"""data = np.array(data)return np.int16(data * (2 ** (width - 1)))def audio_i2f(data, width=16):"""将整数音频数据转换为浮点数音频数据。"""data = np.array(data)return np.float32(data / (2 ** (width - 1)))def save_wavfile(path, wave_data):"""保存音频数据为wav文件。"""with wave.open(path, 'wb') as wav_file:wav_file.setnchannels(1)wav_file.setsampwidth(2)wav_file.setframerate(16000)wav_file.writeframes(np.array(wave_data).tobytes())print(f"Successfully saved wavfile: {path} ..")# 获取麦克风设备列表
def list_devices():p = pyaudio.PyAudio()info = p.get_host_api_info_by_index(0)numdevices = info.get('deviceCount')devices = []for i in range(numdevices):if p.get_device_info_by_host_api_device_index(0, i).get('maxInputChannels') > 0:devices.append(p.get_device_info_by_host_api_device_index(0, i))p.terminate()print("Available recording devices:")devices_dict = {}for i, device in enumerate(devices):print(f"{i}: {device['name']}")devices_dict[device['name']] = ireturn devices, devices_dictdef init_device():devices, devices_dict = list_devices()device_id = int(input("请选择设备:"))print("选择设备：",devices[device_id])device = devices[device_id]  # Select the first available device, modify as neededreturn deviceclass Recorder(threading.Thread):def __init__(self,format=pyaudio.paInt16,channels=1,sample_rate=16000,frames_per_buffer=1024,device = None):super().__init__()self.daemon = Trueself._stop_event = threading.Event()self.device = deviceself.init_stream(format=format,channels=channels,sample_rate=sample_rate,frames_per_buffer=frames_per_buffer)self.waveform = []def run(self):self.waveform = []chunk_size = 1024while not self._stop_event.is_set():data = self.stream.read(chunk_size)data = np.frombuffer(data,dtype='int16')self.waveform.extend(data)self.deinit_stream()def init_stream(self,format=pyaudio.paInt16,channels=1,sample_rate=16000,frames_per_buffer=1024):self.p = pyaudio.PyAudio()self.stream = self.p.open(format=format,channels=channels,rate=sample_rate,input=True,input_device_index=self.device['index'],frames_per_buffer=frames_per_buffer)print("Initialized the stream reader successfully.")def deinit_stream(self):self.stream.stop_stream()self.stream.close()self.p.terminate()print("Deinitialized the stream reader successfully.")def stop(self):self._stop_event.set()if __name__ == "__main__":device = init_device()recorder = Recorder(device=device)input("按下任意键开始录音")recorder.start()input("按下任意键结束录音")recorder.stop()recorder.join()save_wavfile('noise.wav',recorder.waveform)