w3resource

NumPy: Extract all the elements of the first row from a given (4x4) array


Extract First Row of 4x4 Array

Write a NumPy program to extract all the elements of the first row from a given (4x4) array.

Pictorial Presentation:

NumPy: Extract the data of the highlighted part of the array values

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 (first row)
print("\nExtracted data: First row")

# Printing the first row of the 'arra_data' array using slicing
print(arra_data[0]) 

Sample Output:

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

Extracted data: First row
[0 1 2 3]

Explanation:

In the above code -

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[0]): Here print() functiont prints the first row of the array arra_data. In this case, it will print [0, 1, 2, 3]. The index 0 corresponds to the first row in the array.


For more Practice: Solve these Related Problems:

  • Write a NumPy program to extract the first row from a 4x4 array using slicing.
  • Create a function that returns the first row of any given 2D array regardless of its dimensions.
  • Test row extraction using both positive and negative indexing to verify correct results.
  • Implement an alternative approach using np.take to retrieve the first row and compare outputs.

Go to:


PREV : Count Dimensions, Elements, and Bytes of Array
NEXT : Extract Second Row of 4x4 Array


Python-Numpy Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.