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

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