// We then plot the first-order coefficients, which are arranged along time
// and log-frequency.
plt.figure(figsize=(8, 2))
plt.imshow(Sx[order1], aspect="auto")
plt.title("First-order scattering")
After Change
// frequency and one second-order frequency. Here, both indices are mixed along
// the vertical axis.
plt.subplot(3, 1, 3)
plt.imshow(Sx[order2], aspect="auto")
plt.title("Second-order scattering")
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
