Python中的音频频率

Posted 2023-02-25

【中文标题】Python中的音频频率

【英文标题】：Audio Frequencies in Python

【发布时间】：2018-11-14 21:03:36

【问题描述】：

我正在编写一个代码来分析由声音演唱的单个音频频率。我需要一种方法来分析音符的频率。目前我正在使用 PyAudio 录制音频文件，存储为.wav，然后立即播放。

import numpy as np
import pyaudio
import wave

# open up a wave
wf = wave.open('file.wav', 'rb')
swidth = wf.getsampwidth()
RATE = wf.getframerate()
# use a Blackman window
window = np.blackman(chunk)
# open stream
p = pyaudio.PyAudio()
stream = p.open(format =
                p.get_format_from_width(wf.getsampwidth()),
                channels = wf.getnchannels(),
                rate = RATE,
                output = True)

# read some data
data = wf.readframes(chunk)

print(len(data))
print(chunk*swidth)

# play stream and find the frequency of each chunk
while len(data) == chunk*swidth:
    # write data out to the audio stream
    stream.write(data)
    # unpack the data and times by the hamming window
    indata = np.array(wave.struct.unpack("%dh"%(len(data)/swidth),\
                                         data))*window
    # Take the fft and square each value
    fftData=abs(np.fft.rfft(indata))**2
    # find the maximum
    which = fftData[1:].argmax() + 1
    # use quadratic interpolation around the max
    if which != len(fftData)-1:
        y0,y1,y2 = np.log(fftData[which-1:which+2:])
        x1 = (y2 - y0) * .5 / (2 * y1 - y2 - y0)
        # find the frequency and output it
        thefreq = (which+x1)*RATE/chunk
        print("The freq is %f Hz." % (thefreq))
    else:
        thefreq = which*RATE/chunk
        print("The freq is %f Hz." % (thefreq))
    # read some more data
    data = wf.readframes(chunk)
if data:
    stream.write(data)
stream.close()
p.terminate()

问题出在 while 循环上。由于某种原因，该条件永远不会成立。我打印出两个值（len(data) 和 (chunk*swidth))，它们分别是 8192 和 4096。然后我尝试在 while 循环中使用 2*chunk*swidth ，这引发了这个错误：

File "C:\Users\Ollie\Documents\Computing A Level CA\pyaudio test.py", line 102, in <module>
data))*window
ValueError: operands could not be broadcast together with shapes (4096,) (2048,)

【问题讨论】：

Scipy 有信号处理，this answer 讨论其他可能性

二进制、十六进制和十进制都代表同一个东西。 0xA=10=1010。仅通过 FFT 运行您的数据不会为您提供基本频率。声音会产生多个频率，因此您需要进行更多的处理和分析才能获得频率。

【参考方案1】：

下面的这个函数找到频谱。我还包含了一个正弦信号和一个 WAV 文件示例应用程序。这是出于教育目的；您也可以使用现成的matplotlib.pyplot.magnitude_spectrum（见下文）。

from scipy import fft, arange
import numpy as np
import matplotlib.pyplot as plt
from scipy.io import wavfile
import os


def frequency_spectrum(x, sf):
    """
    Derive frequency spectrum of a signal from time domain
    :param x: signal in the time domain
    :param sf: sampling frequency
    :returns frequencies and their content distribution
    """
    x = x - np.average(x)  # zero-centering

    n = len(x)
    k = arange(n)
    tarr = n / float(sf)
    frqarr = k / float(tarr)  # two sides frequency range

    frqarr = frqarr[range(n // 2)]  # one side frequency range

    x = fft(x) / n  # fft computing and normalization
    x = x[range(n // 2)]

    return frqarr, abs(x)


# Sine sample with a frequency of 1hz and add some noise
sr = 32  # sampling rate
y = np.linspace(0, 2*np.pi, sr)
y = np.tile(np.sin(y), 5)
y += np.random.normal(0, 1, y.shape)
t = np.arange(len(y)) / float(sr)

plt.subplot(2, 1, 1)
plt.plot(t, y)
plt.xlabel('t')
plt.ylabel('y')

frq, X = frequency_spectrum(y, sr)

plt.subplot(2, 1, 2)
plt.plot(frq, X, 'b')
plt.xlabel('Freq (Hz)')
plt.ylabel('|X(freq)|')
plt.tight_layout()


# wav sample from https://freewavesamples.com/files/Alesis-Sanctuary-QCard-Crickets.wav
here_path = os.path.dirname(os.path.realpath(__file__))
wav_file_name = 'Alesis-Sanctuary-QCard-Crickets.wav'
wave_file_path = os.path.join(here_path, wav_file_name)
sr, signal = wavfile.read(wave_file_path)

y = signal[:, 0]  # use the first channel (or take their average, alternatively)
t = np.arange(len(y)) / float(sr)

plt.figure()
plt.subplot(2, 1, 1)
plt.plot(t, y)
plt.xlabel('t')
plt.ylabel('y')

frq, X = frequency_spectrum(y, sr)

plt.subplot(2, 1, 2)
plt.plot(frq, X, 'b')
plt.xlabel('Freq (Hz)')
plt.ylabel('|X(freq)|')
plt.tight_layout()

plt.show()

---

您还可以参考SciPy's Fourier Transforms 和Matplotlib's magnitude spectrum plotting 页面了解更多阅读内容和功能。

magspec = plt.magnitude_spectrum(y, sr)  # returns a tuple with the frequencies and associated magnitudes

【讨论】：

嗨，这对正弦波非常有效，但是我需要更改什么才能将输入更改为 wav 文件？我试过用数据数组替换 y，但它似乎不起作用。

您是否首先将数组的均值设置为 0？我继续将这一步移到函数本身。我也有 wav 到数组转换器的代码；很快就会在这里添加。

添加了一个 wav 示例。

谢谢，效果很好。如果要打印出频率，则需要在 X 中找到最大值，然后使用 frq 数组中的相同索引。