w3resource

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

NumPy: Array Object Exercise-141 with Solution

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

Pictorial Presentation:

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

Sample Solution:

Python Code:

import numpy as np
arra_data = np.arange(0,16).reshape((4, 4))
print("Original array:")
print(arra_data)
print("\nExtracted data: All the elements of the second and third columns")
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]]

Python Code Editor:

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

Previous: Write a NumPy program to extract all the elements of the first and fourth columns from a given (4x4) array.
Next: Write a NumPy program to extract all the elements of the first and fourth columns from a given (4x4) array.

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz



Python: Tips of the Day

Set comprehension:

>>> m = {x ** 2 for x in range(5)}
>>> m
{0, 1, 4, 9, 16}