Converting MATLAB code to Python code issues with arrays/matrices and pcolor/imagesc

Question

I may have a strange question, but I hope not.

I have a MATLAB code, which loads a signal, Fourier transforms it, does some extra computations and then plots the resu