- Trending Categories
- Data Structure
- Networking
- RDBMS
- Operating System
- Java
- iOS
- HTML
- CSS
- Android
- Python
- C Programming
- C++
- C#
- MongoDB
- MySQL
- Javascript
- PHP

- Selected Reading
- UPSC IAS Exams Notes
- Developer's Best Practices
- Questions and Answers
- Effective Resume Writing
- HR Interview Questions
- Computer Glossary
- Who is Who

# Extract the diagonal of a matrix with Einstein summation convention in Python

The einsum() method evaluates the Einstein summation convention on the operands. Using the Einstein summation convention, many common multi-dimensional, linear algebraic array operations can be represented in a simple fashion. In implicit mode einsum computes these values.

In explicit mode, einsum provides further flexibility to compute other array operations that might not be considered classical Einstein summation operations, by disabling, or forcing summation over specified subscript labels. To extract the diagonal of a matrix with Einstein summation convention, use the numpy.einsum() method in Python.

The 1st parameter is the subscript. It specifies the subscripts for summation as comma separated list of subscript labels. The 2nd parameter is the operands. These are the arrays for the operation.

## Steps

At first, import the required libraries −

import numpy as np

Creating a numpy array using the arange() and reshape() method −

arr = np.arange(16).reshape(4,4)

Display the array −

print("Our Array...\n",arr)

Check the Dimensions −

print("\nDimensions of our Array...\n",arr.ndim)

Get the Datatype −

print("\nDatatype of our Array object...\n",arr.dtype)

Get the Shape −

print("\nShape of our Array object...\n",arr.shape)

To extract the diagonal of a matrix with Einstein summation convention, use the numpy.einsum() method −

print("\nResult...\n",np.einsum('ii->i', arr))

## Example

import numpy as np # Creating a numpy array using the arange() and reshape() method arr = np.arange(16).reshape(4,4) # Display the array print("Our Array...\n",arr) # Check the Dimensions print("\nDimensions of our Array...\n",arr.ndim) # Get the Datatype print("\nDatatype of our Array object...\n",arr.dtype) # Get the Shape print("\nShape of our Array object...\n",arr.shape) # To extract the diagonal of a matrix with Einstein summation convention, use the numpy.einsum() method in Python. print("\nResult...\n",np.einsum('ii->i', arr))

## Output

Our Array... [[ 0 1 2 3] [ 4 5 6 7] [ 8 9 10 11] [12 13 14 15]] Dimensions of our Array... 2 Datatype of our Array object... int64 Shape of our Array object... (4, 4) Result... [ 0 5 10 15]

- Related Questions & Answers
- Matrix Vector multiplication with Einstein summation convention in Python
- Compute a matrix transpose with Einstein summation convention in Python
- Get the trace of a matrix with Einstein summation convention in Python
- Scalar multiplication with Einstein summation convention in Python
- Tensor contraction with Einstein summation convention in Python
- Array axis summations with Einstein summation convention in Python
- Vector inner product with Einstein summation convention in Python
- Vector outer product with Einstein summation convention in Python
- How to extract diagonal elements of a matrix in R without using diag function?
- Diagonal product of a matrix - JavaScript
- Program to find diagonal sum of a matrix in Python
- Python Program to Extract Rows of a matrix with Even frequency Elements
- Print matrix in diagonal pattern
- Convert a single column matrix into a diagonal matrix in R.
- Python Program to Sort Matrix Rows by summation of consecutive difference of elements