Just plots a wave and its fft

printwave.py

import matplotlib.pyplot as plt
import numpy as np
import wave
import sys
import scipy.fftpack
import os

import argparse

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('f',type=str)
    return parser.parse_args()

args = parse_args()


spf = wave.open(args.f,'r')
fname = os.path.basename(args.f)
bname,_ = os.path.splitext(fname)
#Extract Raw Audio from Wav File
signal = spf.readframes(-1)
signal = np.fromstring(signal, 'Int16')

yf = scipy.fftpack.fft(signal)

#If Stereo
if spf.getnchannels() == 2:
    print 'Just mono files'
    sys.exit(0)

fig, ax = plt.subplots()
fig.patch.set_visible(False)
ax.axis('off')
ax.plot(signal)
fig.savefig(bname+'.png')

fig, ax = plt.subplots()
fig.patch.set_visible(False)
ax.axis('off')
ax.plot(yf[0:signal.shape[0]/2])
fig.savefig(bname+'fft.png')