NumPy N-Dimensional Arrays (ndarray)

NumPy (Numerical Python) is a popular open-source Python library used for scientific computing, numerical calculations, and data analysis. It allows you to create powerful, high-performance multi-dimensional arrays.

Getting started requires importing the library and initializing your first array. Here are the basic code blocks to check your installation and create a simple array:

import numpy as np
print("NumPy Version:", np.__version__)

npr = np.array([4, 5, 9])  
print(npr) # Output: [4 5 9]

Key Features of NumPy

1. Fast vectorized operations: Perform operations on entire arrays rather than looping through elements individually.

   import numpy as np
   print(np.array([1, 2, 3]) * 10) # Output: [10 20 30]

2. Broadcasting: Perform calculations between arrays of different shapes and sizes smoothly.

   import numpy as np
   
   a1 = np.array([2, 4, 6])                # 1D array (shape: (3,))
   a2 = np.array([[1, 3, 5],
                  [7, 9, 11]])             # 2D array (shape: (2,3))
   
   result = a1 + a2
   print(result)
   # Output: [[ 3  7 11]
   #          [ 9 13 17]]

3. Linear algebra: Built-in suite of functions for matrix multiplication, determinants, eigenvalues, and eigenvectors.

   import numpy as np
   A = np.array([[1, 2], [3, 4]])
   B = np.array([[5, 6], [7, 8]])
   result = np.dot(A, B)
   print(result)
   # Output: [[19 22]
   #          [43 50]]

4. Random number generation: Tools for simulating random data distributions and statistical analysis.

   x = np.random.randint(0, 100, 5)
   print(x) # 1D array of 5 random integers between 0 and 100

Installing NumPy

Run the installation command in your terminal or development environment:

pip install numpy

Core NumPy Learning Tracks

Explore our structured reference paths covering array creation, mathematics, random sampling, and performance engineering:

Inspect all functions available in NumPy using dir():

import numpy as np
print(len(dir(np))) # 532

Common Use Cases of NumPy

NumPy is the backbone of scientific and analytical computing in Python. Its fast, vectorized operations and efficient memory management make it indispensable across multiple domains:

• Data Science & Machine Learning:
  • Data cleaning and preprocessing with numerical arrays.
  • Feature engineering and normalization of datasets.
  • Representing datasets as NumPy arrays for machine learning models (e.g., input for scikit-learn).
  • Implementing custom ML algorithms and gradient-based optimizations.
• Scientific Computing:
  • Performing numerical simulations in physics, chemistry, and engineering.
  • Solving systems of linear equations and matrix transformations.
  • Signal and image processing using Fourier transforms.
  • Advanced statistical and probabilistic analysis.
• Image Processing: Images are commonly represented as 2D or 3D NumPy arrays where dimensions correspond to height, width, and color channels, ideal for filtering, masking, and feature extraction.
• Financial Modeling: Simplifies time series analysis, portfolio optimization, and risk assessment through efficient handling of large numerical datasets.

Subhendu Mohapatra

Author & Instructor at plus2net

I write and maintain practical tutorials on Python, PHP, SQL, JavaScript, HTML, jQuery, and web development at plus2net. The tutorials focus on clear explanations, working examples, and code that readers can test and adapt while learning.

About the author   Watch on YouTube

Connect: