Numpy Cheatbook
Numpy Cheatbook
Numpy Cheatbook
Objective:
In this reading, you'll learn:
Basics of NumPy
How to create NumPy arrays
Array attributes and indexing
Basic operations like addition and multiplication
What is NumPy?
NumPy, short for Numerical Python, is a fundamental library for numerical and scientific computing in Python. It provides
support for large, multi-dimensional arrays and matrices, along with a collection of high-level mathematical functions to operate
on these arrays. NumPy serves as the foundation for many data science and machine learning libraries, making it an essential tool
for data analysis and scientific research in Python.
Efficient data structures: NumPy introduces efficient array structures, which are faster and more memory-efficient than
Python lists. This is crucial for handling large data sets.
Multi-dimensional arrays: NumPy allows you to work with multi-dimensional arrays, enabling the representation of
matrices and tensors. This is particularly useful in scientific computing.
Element-wise operations: NumPy simplifies element-wise mathematical operations on arrays, making it easy to perform
calculations on entire data sets in one go.
Random number generation: It provides a wide range of functions for generating random numbers and random data,
which is useful for simulations and statistical analysis.
Integration with other libraries: NumPy seamlessly integrates with other data science libraries like SciPy, Pandas, and
Matplotlib, enhancing its utility in various domains.
Performance optimization: NumPy functions are implemented in low-level languages like C and Fortran, which
significantly boosts their performance. It's a go-to choice when speed is essential.
Installation
If you haven't already installed NumPy, you can do so using pip:
1. 1
Copied!
Creating 1D array
1. 1
about:blank 1/4
13/03/2024, 13:32 about:blank
1. import numpy as np
Copied!
import numpy as np: In this line, the NumPy library is imported and assigned an alias np to make it easier to reference in the
code.
1. 1
2. 2
1. # Creating a 1D array
2. arr_1d = np.array([1, 2, 3, 4, 5]) # **np.array()** is used to create NumPy arrays.
Copied!
arr_1d = np.array([1, 2, 3, 4, 5]): In this line, a one-dimensional NumPy array named arr_1d is created. It uses the np.array()
function to convert a Python list [1, 2, 3, 4, 5] into a NumPy array. This array contains five elements, which are 1, 2, 3, 4, and 5.
arr_1d is a 1D array because it has a single row of elements.
Creating 2D array
1. 1
1. import numpy as np
Copied!
import numpy as np: In this line, the NumPy library is imported and assigned an alias np to make it easier to reference in the
code.
1. 1
2. 2
1. # Creating a 2D array
2. arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
Copied!
arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]): In this line, a two-dimensional NumPy array named arr_2d is created. It uses
the np.array() function to convert a list of lists into a 2D NumPy array.
The outer list contains three inner lists, each of which represents a row of elements. So, arr_2d is a 2D array with three rows and
three columns. The elements in this array form a matrix with values from 1 to 9, organized in a 3x3 grid.
Array attributes
1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
1. # Array attributes
2. print(arr_2d.ndim) # ndim : Represents the number of dimensions or "rank" of the array.
3. # output : 2
4. print(arr_2d.shape) # shape : Returns a tuple indicating the number of rows and columns in the array.
5. # Output : (3, 3)
6. print(arr_2d.size) # size: Provides the total number of elements in the array.
7. # Output : 9
Copied!
In this line, the third element (index 2) of the 1D array arr_1d is accessed.
1. 1
2. 2
about:blank 2/4
13/03/2024, 13:32 about:blank
2. print(arr_1d[2]) # Accessing an element (3rd element)
Copied!
In this line, the element in the 2nd row (index 1) and 3rd column (index 2) of the 2D array arr_2d is accessed.
1. 1
Copied!
In this line, the 2nd row (index 1) of the 2D array arr_2d is accessed.
1. 1
Copied!
In this line, the 2nd column (index 1) of the 2D array arr_2d is accessed.
1. 1
Copied!
Basic operations
NumPy simplifies basic operations on arrays:
Array addition
1. 1
2. 2
3. 3
4. 4
5. 5
1. # Array addition
2. array1 = np.array([1, 2, 3])
3. array2 = np.array([4, 5, 6])
4. result = array1 + array2
5. print(result) # [5 7 9
Copied!
Scalar multiplication
1. 1
2. 2
3. 3
4. 4
1. # Scalar multiplication
2. array = np.array([1, 2, 3])
3. result = array * 2 # each element of an array is multiplied by 2
4. print(result) # [2 4 6]
Copied!
1. 1
2. 2
3. 3
4. 4
5. 5
about:blank 3/4
13/03/2024, 13:32 about:blank
1. # Element-wise multiplication (Hadamard product)
2. array1 = np.array([1, 2, 3])
3. array2 = np.array([4, 5, 6])
4. result = array1 * array2
5. print(result) # [4 10 18]
Copied!
Matrix multiplication
1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
1. # Matrix multiplication
2. matrix1 = np.array([[1, 2], [3, 4]])
3. matrix2 = np.array([[5, 6], [7, 8]])
4. result = np.dot(matrix1, matrix2)
5. print(result)
6. # [[19 22]
7. # [43 50]]
Copied!
NumPy simplifies these operations, making it easier and more efficient than traditional Python lists.
Conclusion
NumPy is a fundamental library for data science and numerical computations. This guide covers the basics of NumPy, and there's
much more to explore. Visit numpy.org for more information and examples.
Author
Akansha Yadav
about:blank 4/4