def spectrogram(data):
import pyqtgraph as pg
import pyqtgraph.exporters
from pyqtgraph.Qt import QtCore, QtGui
app = QtGui.QApplication([])
plot = pg.ImageView(view=pg.PlotItem())
plot.setImage(data)
exporterp = pg.exporters.ImageExporter(plot.imageItem)
exporterp.export('spectrogramp1.png')
exporters = pg.exporters.SVGExporter(plot.imageItem)
exporters.export('spectrograms1.svg')
plot.close()
if __name__ == "__main__":
import numpy as np
test = np.random.randint(-32000, 32000, (2048, 2048))
spectrogram(test)