Create 2D Array in NumPy
NumPy - Create 2D Array
To create a 2D (2 dimensional) array in Python using NumPy library, we can use any of the following methods.
- numpy.array() - Creates array from given values.
- numpy.zeros() - Creates array of zeros.
- numpy.ones() - Creates array of ones.
- numpy.empty() - Creates an empty array.
1. Create 2D Array using numpy.array()
Pass a list of lists to numpy.array() function.
Python Program
import numpy as np
# create a 2D array with shape (3, 4)
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
print(arr)
Output
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
2. Create 2D Array using numpy.zeros()
Pass shape of the required 2D array, as a tuple, as argument to numpy.zeros() function. The function returns a numpy array with specified shape, and all elements in the array initialised to zeros.
Python Program
import numpy as np
# create a 2D array with shape (3, 4)
shape = (3, 4)
arr = np.zeros(shape)
print(arr)
Output
[[0. 0. 0. 0.]
[0. 0. 0. 0.]
[0. 0. 0. 0.]]
3. Create 2D Array using numpy.ones()
Pass shape of the required 2D array, as a tuple, as argument to numpy.ones() function. The function returns a numpy array with specified shape, and all elements in the array initialised to ones.
Python Program
import numpy as np
# create a 2D array with shape (3, 4)
shape = (3, 4)
arr = np.ones(shape)
print(arr)
Output
[[1. 1. 1. 1.]
[1. 1. 1. 1.]
[1. 1. 1. 1.]]
4. Create 2D Array using numpy.empty()
Pass shape of the required 2D array, as a tuple, as argument to numpy.empty() function. The function returns a numpy array with specified shape.
Python Program
import numpy as np
# create a 2D array with shape (3, 4)
shape = (3, 4)
arr = np.empty(shape)
print(arr)
Output
[[0. 0. 0. 0.]
[0. 0. 0. 0.]
[0. 0. 0. 0.]]
Summary
In this NumPy Tutorial, we learned how to create a 2D numpy array in Python using different NumPy functions.