NumPy: Extract all the elements of the second and third columns from a given (4x4) array

Last update on December 21 2024 07:37:48 (UTC/GMT +8 hours)

Extract Second & Third Columns

Write a NumPy program to extract all the elements of the second and third columns from a given (4x4) array.

Pictorial Presentation:

Sample Solution:

Python Code:

# Importing the NumPy library and aliasing it as 'np'
import numpy as np

# Creating a NumPy array 'arra_data' containing integers from 0 to 15 and reshaping it into a 4x4 matrix
arra_data = np.arange(0, 16).reshape((4, 4))

# Displaying a message indicating the original array will be printed
print("Original array:")

# Printing the original 4x4 array 'arra_data'
print(arra_data)

# Displaying a message indicating the extracted data (all the elements of the second and third columns)
print("\nExtracted data: All the elements of the second and third columns")

# Using slicing to extract all rows and specific columns (2nd and 3rd columns) from 'arra_data'
print(arra_data[:, [1, 2]])

Sample Output:

Original array:
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]

Extracted data: All the elements of the second and third columns
[[ 1  2]
 [ 5  6]
 [ 9 10]
 [13 14]]

Explanation:

In the above exercise -

arra_data = np.arange(0, 16).reshape((4, 4)): It creates a 1-dimensional NumPy array with elements from 0 to 15 (excluding 16) using np.arange(0, 16) and then reshapes it into a 2-dimensional array with 4 rows and 4 columns using .reshape((4, 4)).

print(arra_data[:, [1, 2]]): This line prints specific columns of ‘arra_data’ by selecting all rows (indicated by :) and only the columns with indices 1 and 2 (indicated by [1, 2]).

Python-Numpy Code Editor: