numpy resize

numpy.resize(a, new_shape)

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
resized_arr = np.resize(arr, (3, 2))

print(resized_arr)

array([[1, 2],
       [3, 4],
       [5, 6]])

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])
resized_arr = np.resize(arr, (2, 4))

print(resized_arr)

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

import numpy as np

arr = np.array([1, 2, 3])
resized_arr = np.resize(arr, (3, 4))

print(resized_arr)

array([[1, 2, 3, 1],
       [2, 3, 1, 2],
       [3, 1, 2, 3]])

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
resized_arr = np.resize(arr, (2, 2))

print(resized_arr)

array([[1, 2],
       [3, 4]])