Comment by papa2fire

11 seconds in colab implementing the basic transformation with reshape, roll, and flatten

4 seconds in google apps script by handling indices in loops instead of rearranging elements