NumPy: Create an array which looks like below array
Create 2D Array with Specific Values
Write a NumPy program to create an array like the one below.
Sample Solution:
Python Code:
# Importing the NumPy library with an alias 'np'
import numpy as np
# Creating a 4x3 array with values from 2 to 13 and generating an upper triangular matrix excluding the main diagonal (shifted by -1)
x = np.triu(np.arange(2, 14).reshape(4, 3), -1)
# Printing the resulting upper triangular matrix
print(x)
Sample Output:
[[ 2 3 4] [ 5 6 7] [ 0 9 10] [ 0 0 13]]
Explanation:
The above code demonstrates how to create a NumPy array using the np.tri() function.
x = np.tri(4, 3, -1): This line creates a NumPy array ‘x’ using the np.tri() function. The function takes three arguments: the number of rows (4), the number of columns (3), and the diagonal below which the elements should be set to 1 (-1). The array is created with ones at and below the specified diagonal and zeros elsewhere.
Python-Numpy Code Editor: