有没有什么办法，我可以录制来自我的电脑的声音，并将其保存为wav或mp3文件在python？

import pyaudio
import wave

# 定义录音参数
FORMAT = pyaudio.paInt16
CHANNELS = 1
RATE = 44100
CHUNK = 1024
RECORD_SECONDS = 5
WAVE_OUTPUT_FILENAME = "output.wav"

# 初始化PyAudio
audio = pyaudio.PyAudio()

# 打开音频流
stream = audio.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("录音完成！")

# 停止音频流
stream.stop_stream()
stream.close()
audio.terminate()

# 保存音频文件
wave_file = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
wave_file.setnchannels(CHANNELS)
wave_file.setsampwidth(audio.get_sample_size(FORMAT))
wave_file.setframerate(RATE)
wave_file.writeframes(b''.join(frames))
wave_file.close()

print("音频文件保存成功：", WAVE_OUTPUT_FILENAME)

from pydub import AudioSegment

# 加载wav文件
audio = AudioSegment.from_wav('output.wav')

# 保存为mp3文件
audio.export('output.mp3', format='mp3')

print("音频文件保存成功：output.mp3")