NumPy provides a powerful set of linear algebra functions that can be used for a variety of data analysis tasks.
Here are some basic linear algebra operations that can be performed with NumPy:
- Dot product: The dot product of two vectors or matrices can be calculated using the
dot()
function.
import numpy as np # Create two vectors a = np.array([1, 2, 3]) b = np.array([4, 5, 6]) # Calculate the dot product of the two vectors c = np.dot(a, b) print(c) # Create two matrices d = np.array([[1, 2], [3, 4]]) e = np.array([[5, 6], [7, 8]]) # Calculate the dot product of the two matrices f = np.dot(d, e) print(f)
2. Matrix inverse: The inverse of a matrix can be calculated using the inv()
function.
import numpy as np # Create a matrix a = np.array([[1, 2], [3, 4]]) # Calculate the inverse of the matrix b = np.linalg.inv(a) print(b)
3. Eigenvalues and eigenvectors: The eigenvalues and eigenvectors of a matrix can be calculated using the eig()
function.
import numpy as np # Create a matrix a = np.array([[1, 2], [3, 4]]) # Calculate the eigenvalues and eigenvectors of the matrix eigenvalues, eigenvectors = np.linalg.eig(a) print("Eigenvalues:", eigenvalues) print("Eigenvectors:", eigenvectors)
4. Singular value decomposition: The singular value decomposition (SVD) of a matrix can be calculated using the svd()
function.
import numpy as np # Create a matrix a = np.array([[1, 2], [3, 4]]) # Calculate the SVD of the matrix u, s, vh = np.linalg.svd(a) print("U:", u) print("S:", s) print("Vh:", vh)
These are just a few basic linear algebra operations that can be performed with NumPy. NumPy’s linear algebra functions are powerful and flexible, and can be used in a variety of data analysis tasks, such as machine learning, image processing, and scientific computing.