マイク入力音声をリアルタイムでFFTしてグラフ化する

初めに

　この記事はガチでコードだけを置いておこうというやつです。
　なので解説は行っていませんが悪しからず。

コード

import pyaudio
import matplotlib.pyplot as plot
import numpy as np

def audiostart():
    audio = pyaudio.PyAudio()
    stream = audio.open( format = pyaudio.paInt16,
        rate = 44100,
        channels = 1,
        input_device_index = 1,
        input = True,
        frames_per_buffer = 1024)
    return audio, stream

def audiostop(audio, stream):
    stream.stop_stream()
    stream.close()
    audio.terminate()

def read_plot_data(stream):
    data = stream.read(1024)
    ndarray  = np.frombuffer(data, dtype='int16')
    wave_y = np.fft.fft(ndarray)
    wave_y = np.abs(wave_y)
    plot.plot(wave_y)
    plot.draw()
    plot.pause(0.1)
    plot.cla()

if __name__ == '__main__':
    (audio,stream) = audiostart()
    while True:
        try:
            read_plot_data(stream)
        except KeyboardInterrupt:
            break
    audiostop(audio,stream)

参照