大家好!我是新手!Python音频转换为时域和频域

我现在是拿到一段音频,我想把它转成时域和频域图,但是一直报错“ValueError: x and y must have same first dimension, but have shapes (396900,) and (198450,)”,请问这个怎么解决呀?

以下是我的代码:

#读取wav文件的音乐,绘制出波形图和FFT图,并导出数据
import wave
import numpy
from pyaudio import PyAudio
import matplotlib.pyplot as plt

filename = 'QWW_1.WAV'
# 只读方式打开WAV文件
wf = wave.open(filename, 'rb')
# 创建PyAudio对象
p = PyAudio()
stream = p.open(format = p.get_format_from_width(wf.getsampwidth()),
    channels = wf.getnchannels(),
    rate = wf.getframerate(),
    output = True)
nframes = wf.getnframes()
framerate = wf.getframerate()

# 读取完整的帧数据到str_data中,这是一个string类型的数据
str_data = wf.readframes(nframes)
wf.close()

# 将波形数据转换成数组
wave_data = numpy.fromstring(str_data, dtype=numpy.short)



# 将wave_data数组改为2列,行数自动匹配
wave_data.shape = -1,2
# 将数组转置
wave_data = wave_data.T

def time_plt():
    # time也是一个数组,与wave_data[0]或wave_data[1]配对形成系列点坐标
    time = numpy.arange(0, nframes)*(1.0/framerate)
    # 绘制波形图
    plt.subplot(211)
    plt.plot(time, wave_data[0], c='r')
    plt.subplot(212)
    plt.plot(time, wave_data[1], c='g')
    plt.xlabel('time (seconds)')
    f = open("%s.txt"%(filename[:-4]),'w') 
    for i in range(len(wave_data[1])):
        if wave_data[1][i] > 0:
            continue
        f.write('%s \n'%(wave_data[1][i]))
    f.close()
    plt.show()

def freq():
    # 采样点数,修改采样点数和起始位置进行不同位置和长度的音频波形分析
    N = 44100
    start = 0  # 开始采样位置
    df = framerate/(N-1)  # 分辨率
    freq = [df*n for n in range(0, N)]  # N个元素
    wave_data2 = wave_data[0][start:start+N]
    c = numpy.fft.fft(wave_data2)*2/N
    # 常规显示采样频率一半的频谱
    d = int(len(c)/2)
    # 仅显示频率在4000以下的频谱
    while freq[d] > 4000:
        d -= 10
    plt.plot(freq[:d-1], abs(c[:d-1]), 'r')
    f = open("%s-FFT.txt"%(filename[:-4]),'w')
    for i in range(len(freq[:d-1])):
        f.write('%s , %s \n'%(freq[i], abs(c[i])))
    f.close()
    plt.show()

def main():
    time_plt()
    freq()

if __name__ == '__main__':
    main()
  • 11
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值