NumPy is the backbone of numerical computing in Python. It powers everything from simple arrays to complex calculations and machine learning workflows.
Anyone aiming for data science, machine learning, or scientific computing roles really needs to know NumPy well.
I have given 75 commonly asked NumPy interview questions. It ranges from basics like arrays and data types to advanced topics such as broadcasting, memory layout, and performance tricks.
The questions dig into array creation, manipulation, math operations, indexing, and how NumPy connects with other libraries. There are practical explanations of universal functions, vectorisation, linear algebra, and ways to handle complex data structures.
You’ll find key concepts that interviewers love to ask about. Think differences between lists and arrays, handling missing values, masking, fancy indexing, and making operations memory-efficient.
Advanced sections touch on strides, memory mapping, and parallel processing—stuff that separates everyday users from true NumPy pros.
1. What is NumPy, and why is it used?
NumPy stands for Numerical Python. It’s open-source and lets you work with arrays and matrices in Python.
The library is great at handling big, multi-dimensional arrays quickly. It comes with a bunch of math functions that work right on those arrays.
NumPy forms the base for scientific computing in Python. Developers love it because it’s just so much faster than regular Python lists.
Under the hood, NumPy runs optimised C code. That means array operations run much faster, which is a big deal for data science and scientific work.
import numpy as np
# Create a NumPy array
arr = np.array([1, 2, 3, 4, 5])
print(arr)It also has tools for linear algebra, random numbers, and Fourier transforms. These features make it a must-have for libraries like pandas and scikit-learn.
2. Explain ndarray in NumPy.
The ndarray is NumPy’s main data structure. It stands for N-dimensional array and stores elements that are all of the same type.
An ndarray can have any number of dimensions. A 1D array looks like a list, while a 2D array is more like a table.
import numpy as np
# 1D array
array_1d = np.array([1, 2, 3, 4])
# 2D array
array_2d = np.array([[1, 2], [3, 4]])All elements need to be the same type, like all ints or all floats. That’s part of what makes ndarrays faster for math than Python lists.
You can run operations on the whole array at once. No need to write loops just to add or multiply everything.
3. How do you create a NumPy array?
You start by importing NumPy with import numpy as np. The most common way is to use np.array() to turn a Python list into an array.
import numpy as np
arr = np.array([1, 2, 3, 4, 5])NumPy also has functions to make arrays from scratch. np.zeros() gives you an array of zeros, and np.ones() gives you ones.
zeros_arr = np.zeros((3, 3))
ones_arr = np.ones((2, 4))If you want a sequence of numbers, np.arange() works a lot like Python’s range. np.linspace() makes evenly spaced numbers between two values.
range_arr = np.arange(0, 10, 2)
linspace_arr = np.linspace(0, 1, 5)
You can set the data type at creation with the dtype parameter if you need something specific.
4. Difference between Python lists and NumPy arrays
Python lists can hold anything—strings, numbers, objects. They’re flexible and good for general use.
NumPy arrays only hold one data type and keep everything in a single, continuous block of memory. This makes them way faster for math and scientific work.
import numpy as np
# Python list
python_list = [1, 2, 3, 4, 5]
# NumPy array
numpy_array = np.array([1, 2, 3, 4, 5])Lists take up more memory because they store references. NumPy arrays store the actual values, so they’re more efficient.
With NumPy, you can do vector operations—process everything at once. Lists need you to loop or use a comprehension for that.
# NumPy vector operation
result = numpy_array * 2 # [2, 4, 6, 8, 10]
# List requires a loop or comprehension
result = [x * 2 for x in python_list]5. What is broadcasting in NumPy?
Broadcasting is how NumPy handles operations on arrays with different shapes. It lets you do element-wise math without making extra copies or writing loops.
If two arrays have different shapes, NumPy stretches the smaller one to match the bigger one. It follows certain rules to decide if the shapes are compatible.