How to Reshape NumPy Array Into Any Shape


In this example, we will learn how to reshape the NumPy array into any shape in Python.

Source Code

import numpy as np

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

# In both shapes, the elements required to reshape must be equal.
reshape_n_1 = n.reshape(3, 3)
print(reshape_n_1)

# The elements required to reshape are not equal, it will raise an error.
reshape_n_2 = n.reshape(3, 4)
print(reshape_n_2)

Output:

[[1 2 3]
 [4 5 6]
 [7 8 9]]
Traceback (most recent call last):
  File "demo.py", line 10, in 
    reshape_n_2 = n.reshape(3, 4)
ValueError: cannot reshape array of size 9 into shape (3,4)
Subscribe
Notify of
guest
0 Comments
Inline Feedbacks
View all comments