This is a package for calculating the short time fourier transform (spectrogram) or any other transform in a lapped and windowed fashion.
You can install this library using pip:
pip install stft
Loading a file and calculating the spectrogram, its inverse and saving the result.
import stft
import scipy.io.wavfile as wav
fs, audio = wav.read('input.wav')
specgram = stft.spectrogram(audio)
output = stft.ispectrogram(specgram)
wav.write('output.wav', fs, output)