How to capture the whole real time CSI wave into a figure

[olliec28]

I want to save the whole of the real time CSI wave into a file. I have found the plt.savefig('filename') method. However, this only captures the last bit of the wave at the point where I stop the CSI data collection. Anyone know a way round this where I can capture the whole wave and not just the last part?

Thanks