numpy.diag() in Python

numpy.diag() in Python

Last Updated : 09 Mar, 2022

numpy.diag(a, k=0) : Extracts and construct a diagonal array
Parameters :

a : array_like 
k : [int, optional, 0 by default]
          Diagonal we require; k>0 means diagonal above main diagonal or vice versa.

Returns :

ndarray

Python

# Python Programming illustrating
# numpy.diag method

import numpy as geek

# matrix creation by array input
a = geek.matrix([[1, 21, 30], 
                 [63 ,434, 3], 
                 [54, 54, 56]])

print("Main Diagonal elements : \n", geek.diag(a), "\n")

print("Diagonal above main diagonal : \n", geek.diag(a, 1), "\n")

print("Diagonal below main diagonal : \n", geek.diag(a, -1))

Output :

Main Diagonal elements : 
 [  1 434  56] 

Diagonal above main diagonal : 
 [21  3] 

Diagonal below main diagonal : 
 [63 54]

References :
https://docs.scipy.org/doc/numpy/reference/generated/numpy.diagflat.html#numpy.diagflat
Note :
These NumPy-Python programs won't run on online IDE's, so run them on your systems to explore them
.

numpy.diag() in Python

M

Mohit Gupta_OMG

Improve

Article Tags :

Practice Tags :

python

Similar Reads

numpy.diagflat() in Python

numpy.diagflat (a, k = 0): Create a two-dimensional array with the array_like input as a diagonal to the new output array. Parameters : a : array_like input data with diagonal elements strong>k : [int, optional, 0 by default] Diagonal we require; k>0 means diagonal above main diagonal or vice

numpy.gcd() in Python

numpy.gcd(arr1, arr2, out = None, where = True, casting = â€˜same_kindâ€™, order = â€˜Kâ€™, dtype = None) : This mathematical function helps user to calculate GCD value of |arr1| and |arr2| elements. Greatest Common Divisor (GCD) of two or more numbers, which are not all zero, is the largest positive number

numpy.all() in Python

The numpy.all() function tests whether all array elements along the mentioned axis evaluate to True. Syntax: numpy.all(array, axis = None, out = None, keepdims = class numpy._globals._NoValue at 0x40ba726c) Parameters :Â array :[array_like]Input array or object whose elements, we need to test. axis

numpy.dot() in Python

numpy.dot(vector_a, vector_b, out = None) returns the dot product of vectors a and b. It can handle 2D arrays but considers them as matrix and will perform matrix multiplication. For N dimensions it is a sum-product over the last axis of a and the second-to-last of b : dot(a, b)[i,j,k,m] = sum(a[i,j

numpy.any() in Python

The numpy.any() function tests whether any array elements along the mentioned axis evaluate to True. Syntax :Â numpy.any(a, axis = None, out = None, keepdims = class numpy._globals._NoValue at 0x40ba726c) Parameters :Â array :[array_like]Input array or object whose elements, we need to test. axis :

numpy.alen() in Python

numpy.alen() function is used to return the length of the first dimension of the input array. Syntax : numpy.alen(arr) Parameters : arr : [array_like] Input array. Return : [int]Length of the first dimension of arr. Code #1 : Python3 # Python program explaining # alen() function import numpy as geek

numpy.ceil() in Python

The numpy.ceil() is a mathematical function that returns the ceil of the elements of array. The ceil of the scalar x is the smallest integer i, such that i >= x Syntax : numpy.ceil(x[, out]) = ufunc â€˜ceilâ€™) Parameters : a : [array_like] Input array Return : The ceil of each element with float dat

numpy.diag_indices() in Python

The numpy.diag_indices() function returns indices in order to access the elements of main diagonal of a array with minimum dimension = 2. Returns indices in the form of tuple.Â to access the main diagonal of an array.Â Syntax: numpy.diag_indices(n, n_dim = 2) Parameters :Â n : size of array, for whic

NumPy Array in Python

NumPy (Numerical Python) is a powerful library for numerical computations in Python. It is commonly referred to multidimensional container that holds the same data type. It is the core data structure of the NumPy library and is optimized for numerical and scientific computation in Python. Table of C

numpy.isnan() in Python

The numpy.isnan() function tests element-wise whether it is NaN or not and returns the result as a boolean array. Syntax :Â numpy.isnan(array [, out]) Parameters :Â array : [array_like]Input array or object whose elements, we need to test for infinity out : [ndarray, optional]Output array placed wit